Home
last modified time | relevance | path

Searched defs:ih_end (Results 1 – 23 of 23) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dnhwc_pooling.cpp271 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 Dnchw_pooling.cpp116 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 Dref_pooling.cpp141 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 Dgemm_convolution_utils.cpp610 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 Dnhwc_pooling.cpp271 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 Dnchw_pooling.cpp119 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 Dref_pooling.cpp149 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 Dgemm_convolution_utils.cpp611 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 Dpool.hpp228 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 Dpool.hpp225 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 Dtest_pooling_backward.cpp208 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 Dtest_pooling_backward.cpp208 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 Djit_avx2_convolution.cpp275 int ih_end = nstl::min(jcp.ih, ih_start + ih_block_size); in execute_backward_data() local
H A Djit_brgemm_conv.cpp727 int id_start {0}, id_end {0}, ih_start {0}, ih_end {0}; in maybe_conv_inp() local
H A Djit_avx512_core_bf16_convolution.cpp1058 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
H A Djit_avx512_core_amx_convolution.cpp1119 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
H A Djit_avx512_common_convolution.cpp1552 int ih_start {0}, ih_end {0}; in compute_diff_weights() local
/dports/math/onednn/oneDNN-2.5.1/src/cpu/x64/
H A Djit_avx2_convolution.cpp275 int ih_end = nstl::min(jcp.ih, ih_start + ih_block_size); in execute_backward_data() local
H A Djit_brgemm_conv.cpp883 int id_start {0}, id_end {0}, ih_start {0}, ih_end {0}; in maybe_conv_inp() local
H A Djit_avx512_core_amx_convolution.cpp1100 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
H A Djit_avx512_core_bf16_convolution.cpp1058 auto tr_src_off = [&](int g, int ic, int ih_end, int ij) { in compute_diff_weights_2d()
H A Djit_avx512_common_convolution.cpp1552 int ih_start {0}, ih_end {0}; in compute_diff_weights() local
/dports/math/ideep/ideep-2.0.0-119-gb57539e/tests/
H A Dtest_ideep_common.hpp687 auto ih_end = in check_pool_fwd() local
798 auto ih_end = in check_pool_bwd() local