Home
last modified time | relevance | path

Searched refs:kd_start (Results 1 – 6 of 6) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/aarch64/
H A Djit_uni_i8i8_pooling.cpp873 const int kd_start in execute_forward() local
891 p.kd_range = (size_t)(kd_end - kd_start); in execute_forward()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/aarch64/
H A Djit_uni_i8i8_pooling.cpp873 const int kd_start in execute_forward() local
891 p.kd_range = (size_t)(kd_end - kd_start); in execute_forward()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/
H A Djit_uni_i8i8_pooling.cpp1457 const int kd_start in execute_forward() local
1475 p.kd_range = (size_t)(kd_end - kd_start); in execute_forward()
H A Djit_avx512_common_convolution.cpp1384 int kd_start = nstl::max( in compute_diff_weights_nxc() local
1405 for_(int i_kd = kd_start; i_kd <= kd_end; i_kd++) in compute_diff_weights_nxc()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_uni_i8i8_pooling.cpp1458 dim_t kd_start in execute_forward() local
1477 p.kd_range = kd_end - kd_start; in execute_forward()
H A Djit_avx512_common_convolution.cpp1384 int kd_start = nstl::max( in compute_diff_weights_nxc() local
1405 for_(int i_kd = kd_start; i_kd <= kd_end; i_kd++) in compute_diff_weights_nxc()