| /dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
| H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 118 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
187 offset = dw_weights_d.size() - dw_weights_d.additional_buffer_size(); in execute_forward_thr()
392 const auto wht_h_stride = dw_weights_d.blk_off(0, 0, 0, 1); in execute_forward_thr()
403 = weights_dw + dw_weights_d.blk_off(ocb, 0) + wei_stride; in execute_forward_thr()
|
| H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 122 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
188 offset = dw_weights_d.size() - dw_weights_d.additional_buffer_size(); in execute_forward_thr()
391 const auto wht_h_stride = dw_weights_d.blk_off(0, 0, 0, 1); in execute_forward_thr()
402 = weights_dw + dw_weights_d.blk_off(ocb, 0) + wei_stride; in execute_forward_thr()
|
| H A D | jit_sse41_1x1_convolution.cpp | 74 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
236 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| H A D | jit_avx2_1x1_convolution.cpp | 91 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
279 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| H A D | jit_avx512_common_1x1_convolution.cpp | 93 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
353 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| H A D | jit_avx512_core_bf16_1x1_convolution.cpp | 135 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
364 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| /dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
| H A D | jit_uni_x8s8s32x_1x1_convolution.cpp | 120 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
188 offset = dw_weights_d.size() - dw_weights_d.additional_buffer_size(); in execute_forward_thr()
395 const auto wht_h_stride = dw_weights_d.blk_off(0, 0, 0, 1); in execute_forward_thr()
407 = weights_dw + dw_weights_d.blk_off(ocb, 0) + wei_stride; in execute_forward_thr()
|
| H A D | jit_avx512_core_x8s8s32x_1x1_convolution.cpp | 115 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
187 offset = dw_weights_d.size() - dw_weights_d.additional_buffer_size(); in execute_forward_thr()
395 const auto wht_h_stride = dw_weights_d.blk_off(0, 0, 0, 1); in execute_forward_thr()
407 = weights_dw + dw_weights_d.blk_off(ocb, 0) + wei_stride; in execute_forward_thr()
|
| H A D | jit_sse41_1x1_convolution.cpp | 74 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
236 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| H A D | jit_avx2_1x1_convolution.cpp | 91 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
279 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| H A D | jit_avx512_common_1x1_convolution.cpp | 93 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
353 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|
| H A D | jit_avx512_core_bf16_1x1_convolution.cpp | 135 const memory_desc_wrapper dw_weights_d( in execute_forward_thr() local
364 = &weights_dw[dw_weights_d.blk_off(ch, 0, 0, kh, kw)]; in execute_forward_thr()
|