Searched refs:key_bnorm_tmp_var (Results 1 – 12 of 12) sorted by relevance
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | nspc_batch_normalization.hpp | 80 key_bnorm_tmp_var, stats_buf_sz); in init_scratchpad()
|
H A D | ncsp_batch_normalization.hpp | 79 key_bnorm_tmp_var, C()); in init_scratchpad()
|
H A D | nspc_batch_normalization.cpp | 60 auto tmp_var = scratchpad.template get<acc_data_t>(key_bnorm_tmp_var); in execute_forward()
|
H A D | ncsp_batch_normalization.cpp | 73 variance = scratchpad.template get<acc_data_t>(key_bnorm_tmp_var); in execute_forward()
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | nspc_batch_normalization.hpp | 82 key_bnorm_tmp_var, stats_buf_sz); in init_scratchpad()
|
H A D | ncsp_batch_normalization.hpp | 82 key_bnorm_tmp_var, C()); in init_scratchpad()
|
H A D | nspc_batch_normalization.cpp | 60 auto tmp_var = scratchpad.template get<acc_data_t>(key_bnorm_tmp_var); in execute_forward()
|
H A D | ncsp_batch_normalization.cpp | 73 variance = scratchpad.template get<acc_data_t>(key_bnorm_tmp_var); in execute_forward()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
H A D | gen9_batch_normalization.cpp | 258 scratchpad.book(key_bnorm_tmp_var, conf.ic, in init_scratchpad() 303 key_bnorm_tmp_var); in execute_forward()
|
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/ |
H A D | gen9_batch_normalization.cpp | 263 scratchpad.book(key_bnorm_tmp_var, conf.ic, in init_scratchpad() 308 key_bnorm_tmp_var); in execute_forward()
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/common/ |
H A D | memory_tracking.hpp | 158 key_bnorm_tmp_var, enumerator
|
/dports/math/onednn/oneDNN-2.5.1/src/common/ |
H A D | memory_tracking.hpp | 158 key_bnorm_tmp_var, enumerator
|