Searched refs:reg_ptr_diff_gamma_ (Results 1 – 2 of 2) sorted by relevance
| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_uni_tbb_batch_normalization.cpp | 1318 const Reg64 ®_ptr_diff_gamma_ = r11; member
1364 mov(reg_ptr_diff_gamma_, PARAM_PTR(diff_gamma)); in load_common_params()
1389 vmmword[reg_ptr_diff_gamma_ + reg_off_c_], vzero_); in zeroise()
1394 vmmword[reg_ptr_diff_gamma_ + reg_off_c_ + vlen / 2], in zeroise()
1523 vtmp_, vmmword[reg_ptr_diff_gamma_ + reg_off_c_ + off]); in store_diff_beta_and_diff_gamma()
1526 vmmword[reg_ptr_diff_gamma_ + reg_off_c_ + off], in store_diff_beta_and_diff_gamma()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_tbb_batch_normalization.cpp | 1316 const Reg64 ®_ptr_diff_gamma_ = r11; member
1362 mov(reg_ptr_diff_gamma_, PARAM_PTR(diff_gamma)); in load_common_params()
1387 vmmword[reg_ptr_diff_gamma_ + reg_off_c_], vzero_); in zeroise()
1392 vmmword[reg_ptr_diff_gamma_ + reg_off_c_ + vlen / 2], in zeroise()
1521 vtmp_, vmmword[reg_ptr_diff_gamma_ + reg_off_c_ + off]); in store_diff_beta_and_diff_gamma()
1524 vmmword[reg_ptr_diff_gamma_ + reg_off_c_ + off], in store_diff_beta_and_diff_gamma()
|