/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/gpu/ocl/ |
H A D | ref_pooling.cl | 112 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 D | gen9_pooling.cl | 155 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 D | ref_pooling.cl | 114 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 D | gen9_pooling.cl | 152 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 D | nchw_pooling.cpp | 122 * (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 D | ref_pooling.cpp | 142 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 D | nhwc_pooling.cpp | 272 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 D | gemm_convolution_utils.cpp | 612 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 D | nchw_pooling.cpp | 120 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 D | ref_pooling.cpp | 150 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 D | nhwc_pooling.cpp | 272 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 D | gemm_convolution_utils.cpp | 613 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 D | pool.hpp | 229 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 D | pool.hpp | 226 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 D | test_ideep_common.hpp | 689 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 D | test_pooling_backward.cpp | 210 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 D | test_pooling_backward.cpp | 210 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()
|