Home
last modified time | relevance | path

Searched defs:wei_reduction (Results 1 – 10 of 10) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dgemm_convolution.cpp600 auto wei_reduction in execute_backward_weights_nspc() local
769 auto wei_reduction in execute_backward_weights_ncsp() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dgemm_convolution.cpp597 auto wei_reduction in execute_backward_weights_nspc() local
766 auto wei_reduction in execute_backward_weights_ncsp() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Dgemm_bf16_convolution.cpp974 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_nspc() local
1180 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_ncsp() local
H A Djit_avx512_common_1x1_convolution.cpp651 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local
H A Djit_avx512_core_bf16_1x1_convolution.cpp684 auto wei_reduction = scratchpad.template get<float>(key_conv_wei_reduction); in execute_backward_weights() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Dgemm_bf16_convolution.cpp965 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_nspc() local
1171 auto wei_reduction = ctx.get_scratchpad_grantor().template get<acc_data_t>( in execute_backward_weights_ncsp() local
H A Djit_avx512_core_bf16_1x1_convolution.cpp687 auto wei_reduction = scratchpad.template get<float>(key_conv_wei_reduction); in execute_backward_weights() local
H A Djit_avx512_common_1x1_convolution.cpp653 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_sve_512_1x1_convolution.cpp486 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_sve_512_1x1_convolution.cpp486 auto wei_reduction = scratchpad.get<data_t>(key_conv_wei_reduction); in execute_backward_weights() local