/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | nhwc_pooling.cpp | 271 const auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 443 const auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 602 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local 758 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local
|
H A D | nchw_pooling.cpp | 116 const auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 268 auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 418 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local 575 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local
|
H A D | ref_pooling.cpp | 141 auto ih_end = oh * SH - padT + (KH - 1) * DH + KH; in execute_forward() local 286 auto ih_end = oh * SH - padT + (KH - 1) * DH + KH; in execute_backward() local
|
H A D | gemm_convolution_utils.cpp | 610 const int ih_end = saturate(0, jcp.ih, hp + hb + jcp.kh); in im2col_dt() local
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | nhwc_pooling.cpp | 271 const auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 444 const auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 603 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local 760 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local
|
H A D | nchw_pooling.cpp | 119 const auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 274 auto ih_end = min(oh * SH - padT + KH, IH); in execute_forward() local 426 dim_t ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local 583 auto ih_end = min(oh * SH - padT + KH, IH); in execute_backward() local
|
H A D | ref_pooling.cpp | 149 auto ih_end = oh * SH - padT + (KH - 1) * DH + KH; in execute_forward() local 280 auto ih_end = oh * SH - padT + (KH - 1) * DH + KH; in execute_backward() local
|
H A D | gemm_convolution_utils.cpp | 611 const dim_t ih_end = saturate(dim_t(0), jcp.ih, hp + hb + jcp.kh); in im2col_dt() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/pool/ |
H A D | pool.hpp | 228 auto ih_end = h * SH - PH + (KH - 1) * DH + KH; in get_num_summands() local
|
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/pool/ |
H A D | pool.hpp | 225 auto ih_end = h * SH - PH + (KH - 1) * DH + KH; in get_num_summands() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/ |
H A D | test_pooling_backward.cpp | 208 auto ih_end = oh * pd.strh - pd.padt + (pd.kh - 1) * pd.dh in check_pool_bwd() local
|
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/ |
H A D | test_pooling_backward.cpp | 208 auto ih_end = oh * pd.strh - pd.padt + (pd.kh - 1) * pd.dh in check_pool_bwd() local
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/x64/ |
H A D | jit_avx2_convolution.cpp | 275 int ih_end = nstl::min(jcp.ih, ih_start + ih_block_size); in execute_backward_data() local
|
H A D | jit_brgemm_conv.cpp | 727 int id_start {0}, id_end {0}, ih_start {0}, ih_end {0}; in maybe_conv_inp() local
|
H A D | jit_avx512_core_bf16_convolution.cpp | 1058 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
|
H A D | jit_avx512_core_amx_convolution.cpp | 1119 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
|
H A D | jit_avx512_common_convolution.cpp | 1552 int ih_start {0}, ih_end {0}; in compute_diff_weights() local
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/ |
H A D | jit_avx2_convolution.cpp | 275 int ih_end = nstl::min(jcp.ih, ih_start + ih_block_size); in execute_backward_data() local
|
H A D | jit_brgemm_conv.cpp | 883 int id_start {0}, id_end {0}, ih_start {0}, ih_end {0}; in maybe_conv_inp() local
|
H A D | jit_avx512_core_amx_convolution.cpp | 1100 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
|
H A D | jit_avx512_core_bf16_convolution.cpp | 1058 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
|
H A D | jit_avx512_common_convolution.cpp | 1552 int ih_start {0}, ih_end {0}; in compute_diff_weights() local
|
/dports/math/ideep/ideep-2.0.0-119-gb57539e/tests/ |
H A D | test_ideep_common.hpp | 687 auto ih_end = in check_pool_fwd() local 798 auto ih_end = in check_pool_bwd() local
|