Home
last modified time | relevance | path

Searched refs:iw_end (Results 1 – 17 of 17) sorted by relevance

/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/
H A Dref_pooling.cl112 const int iw_end = ow * SW - PW + (KW - 1) * DW + KW;
125 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0;
235 const int iw_end
253 const int iw_end_excluded = iw_end > IW
254 ? (iw_end - IW - 1) / (DW + 1) + 1
298 const int iw_end = ow * SW - PW + (KW - 1) * DW + KW;
311 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0;
323 for (int iw = iw_start; iw < min(iw_end, IW);
H A Dgen9_pooling.cl155 const int iw_end = min(ow * SW - PW + KW, IW);
156 const DATA_T num_summands = (ih_end - ih_start) * (iw_end - iw_start)
311 const int iw_end = min(iw - kw + KW, IW);
313 * (iw_end - iw_start) * (id_end - id_start);
/dports/math/onednn/oneDNN-2.5.1/src/gpu/ocl/
H A Dref_pooling.cl114 const int iw_end = ow * SW - PW + (KW - 1) * DW + KW;
127 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0;
236 const int iw_end
254 const int iw_end_excluded = iw_end > IW
255 ? (iw_end - IW - 1) / (DW + 1) + 1
299 const int iw_end = ow * SW - PW + (KW - 1) * DW + KW;
312 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0;
324 for (int iw = iw_start; iw < min(iw_end, IW);
H A Dgen9_pooling.cl152 const int iw_end = min(ow * SW - PW + KW, IW);
154 = (ih_end - ih_start) * (iw_end - iw_start) * (id_end - id_start);
313 const int iw_end = min(iw - kw + KW, IW);
315 * (iw_end - iw_start) * (id_end - id_start);
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/
H A Dnchw_pooling.cpp122 * (iw_end - iw_start); in execute_forward()
127 for (int iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
269 auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
274 * (iw_end - iw_start); in execute_forward()
278 for (int iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
419 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
424 * (iw_end - iw_start); in execute_backward()
428 for (int iw = iw_start; iw < iw_end; ++iw) { in execute_backward()
576 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
581 * (iw_end - iw_start); in execute_backward()
[all …]
H A Dref_pooling.cpp142 auto iw_end = ow * SW - padL + (KW - 1) * DW + KW; in execute_forward() local
155 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0; in execute_forward()
287 auto iw_end = ow * SW - padL + (KW - 1) * DW + KW; in execute_backward() local
300 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0; in execute_backward()
H A Dnhwc_pooling.cpp272 const auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
280 for (int iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
444 const auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
452 for (int iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
603 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
608 : (ih_end - ih_start) * (iw_end - iw_start) in execute_backward()
759 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
764 : (ih_end - ih_start) * (iw_end - iw_start) in execute_backward()
H A Dgemm_convolution_utils.cpp612 const int iw_end = saturate(0, jcp.iw, wp + wb + jcp.kw); in im2col_dt() local
615 const int iwb = iw_end - iw_start; in im2col_dt()
624 for (int iw = iw_start; iw < iw_end; iw++) in im2col_dt()
/dports/math/onednn/oneDNN-2.5.1/src/cpu/
H A Dnchw_pooling.cpp120 const auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
125 * (iw_end - iw_start); in execute_forward()
130 for (dim_t iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
275 auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
280 * (iw_end - iw_start); in execute_forward()
284 for (dim_t iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
427 dim_t iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
432 * (iw_end - iw_start); in execute_backward()
436 for (dim_t iw = iw_start; iw < iw_end; ++iw) { in execute_backward()
584 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
[all …]
H A Dref_pooling.cpp150 auto iw_end = ow * SW - padL + (KW - 1) * DW + KW; in execute_forward() local
163 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0; in execute_forward()
281 auto iw_end = ow * SW - padL + (KW - 1) * DW + KW; in execute_backward() local
294 = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0; in execute_backward()
H A Dnhwc_pooling.cpp272 const auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
280 for (dim_t iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
445 const auto iw_end = min(ow * SW - padL + KW, IW); in execute_forward() local
453 for (dim_t iw = iw_start; iw < iw_end; ++iw) { in execute_forward()
604 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
609 : (ih_end - ih_start) * (iw_end - iw_start) in execute_backward()
761 auto iw_end = min(ow * SW - padL + KW, IW); in execute_backward() local
766 : (ih_end - ih_start) * (iw_end - iw_start) in execute_backward()
H A Dgemm_convolution_utils.cpp613 const dim_t iw_end = saturate(dim_t(0), jcp.iw, wp + wb + jcp.kw); in im2col_dt() local
616 const dim_t iwb = iw_end - iw_start; in im2col_dt()
625 for (dim_t iw = iw_start; iw < iw_end; iw++) in im2col_dt()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/benchdnn/pool/
H A Dpool.hpp229 auto iw_end = w * SW - PW + (KW - 1) * DW + KW; in get_num_summands() local
239 auto iw_end_excluded = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0; in get_num_summands()
/dports/math/onednn/oneDNN-2.5.1/tests/benchdnn/pool/
H A Dpool.hpp226 auto iw_end = w * SW - PW + (KW - 1) * DW + KW; in get_num_summands() local
236 auto iw_end_excluded = iw_end > IW ? (iw_end - IW - 1) / (DW + 1) + 1 : 0; in get_num_summands()
/dports/math/ideep/ideep-2.0.0-119-gb57539e/tests/
H A Dtest_ideep_common.hpp689 auto iw_end = in check_pool_fwd() local
694 ? pd.kw*pd.kh : (ih_end - ih_start)*(iw_end - iw_start); in check_pool_fwd()
697 for (int iw = iw_start; iw < iw_end; ++iw) { in check_pool_fwd()
800 auto iw_end = in check_pool_bwd() local
804 ? pd.kw*pd.kh : (ih_end - ih_start)*(iw_end - iw_start); in check_pool_bwd()
807 for (int iw = iw_start; iw < iw_end; iw++) { in check_pool_bwd()
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/tests/gtests/
H A Dtest_pooling_backward.cpp210 auto iw_end = ow * pd.strw - pd.padl + (pd.kw - 1) * pd.dw in check_pool_bwd() local
228 auto iw_end_excluded = iw_end > pd.iw in check_pool_bwd()
229 ? (iw_end - pd.iw - 1) / (pd.dw + 1) + 1 in check_pool_bwd()
/dports/math/onednn/oneDNN-2.5.1/tests/gtests/
H A Dtest_pooling_backward.cpp210 auto iw_end = ow * pd.strw - pd.padl + (pd.kw - 1) * pd.dw in check_pool_bwd() local
228 auto iw_end_excluded = iw_end > pd.iw in check_pool_bwd()
229 ? (iw_end - pd.iw - 1) / (pd.dw + 1) + 1 in check_pool_bwd()