/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | pad.cu | 342 int iStartZ = max(0, -padF); in image_3d_pad_edge_kernel() 345 int oStartZ = max(0, padF); in image_3d_pad_edge_kernel() 362 const int padF = pad[4]; in image_pad_edge() local 395 int oStartZ = max(0, padF); in image_3d_pad_edge_grad_kernel() 413 const int padF = pad[4]; in image_pad_edge_grad() local 472 const int padF = pad[4]; in image_pad_constant() local 514 const int padF = pad[4]; in image_pad_constant_grad() local 551 int oStartZ = max(0, padF); in image_3d_pad_reflect_kernel() 579 const int padF = pad[4]; in image_pad_reflect() local 612 int oStartZ = max(0, padF); in image_3d_pad_reflect_grad_kernel() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/src/operator/ |
H A D | pad.cu | 342 int iStartZ = max(0, -padF); in image_3d_pad_edge_kernel() 345 int oStartZ = max(0, padF); in image_3d_pad_edge_kernel() 362 const int padF = pad[4]; in image_pad_edge() local 395 int oStartZ = max(0, padF); in image_3d_pad_edge_grad_kernel() 413 const int padF = pad[4]; in image_pad_edge_grad() local 472 const int padF = pad[4]; in image_pad_constant() local 514 const int padF = pad[4]; in image_pad_constant_grad() local 551 int oStartZ = max(0, padF); in image_3d_pad_reflect_kernel() 579 const int padF = pad[4]; in image_pad_reflect() local 612 int oStartZ = max(0, padF); in image_3d_pad_reflect_grad_kernel() [all …]
|
/dports/misc/mxnet/incubator-mxnet-1.9.0/3rdparty/mkldnn/src/cpu/ |
H A D | ref_pooling.cpp | 75 const int padF = pd()->padFront(); in execute_forward() local 97 const int id = od * SD - padF + kd * (DD + 1); in execute_forward() 119 const int id = od * SD - padF + kd * (DD + 1); in execute_forward() 137 auto id_start = od * SD - padF; in execute_forward() 140 auto id_end = od * SD - padF + (KD - 1) * DD + KD; in execute_forward() 235 const int padF = pd()->padFront(); in execute_backward() local 260 const int id = od * SD - padF + kd * (DD + 1); in execute_backward() 282 auto id_start = od * SD - padF; in execute_backward() 285 auto id_end = od * SD - padF + (KD - 1) * DD + KD; in execute_backward() 307 const int id = od * SD - padF + kd * (DD + 1); in execute_backward() [all …]
|
H A D | nhwc_pooling.cpp | 184 const int padF = pd()->padFront(); in execute_forward() local 237 const int id = od * SD - padF + kd; in execute_forward() 351 const int padF = pd()->padFront(); in execute_forward() local 408 const int id = od * SD - padF + kd; in execute_forward() 513 const int padF = pd()->padFront(); in execute_backward() local 547 int od_left = max((id + padF - KD + 1) / SD, 0); in execute_backward() 552 int od_right = min((id + padF) / SD + 1, OD); in execute_backward() 559 const int kd = id - od * SD + padF; in execute_backward() 601 auto id_end = min(od * SD - padF + KD, ID); in execute_backward() 657 const int padF = pd()->padFront(); in execute_backward() local [all …]
|
H A D | nchw_pooling.cpp | 64 const int padF = pd()->padFront(); in execute_forward() local 92 const int id = od * SD - padF + kd; in execute_forward() 210 const int padF = pd()->padFront(); in execute_forward() local 243 const int id = od * SD - padF + kd; in execute_forward() 267 auto id_end = min(od * SD - padF + KD, ID); in execute_forward() 361 const int padF = pd()->padFront(); in execute_backward() local 395 const int id = od * SD - padF + kd; in execute_backward() 417 auto id_end = min(od * SD - padF + KD, ID); in execute_backward() 443 int od_end = min(OD, 1 + (padF + ID - 1) / SD); in execute_backward() 514 const int padF = pd()->padFront(); in execute_backward() local [all …]
|
/dports/math/onednn/oneDNN-2.5.1/src/cpu/ |
H A D | ref_pooling.cpp | 79 const dim_t padF = pd()->padFront(); in execute_forward() local 104 const dim_t id = od * SD - padF + kd * (DD + 1); in execute_forward() 127 const dim_t id = od * SD - padF + kd * (DD + 1); in execute_forward() 145 auto id_start = od * SD - padF; in execute_forward() 148 auto id_end = od * SD - padF + (KD - 1) * DD + KD; in execute_forward() 230 const dim_t padF = pd()->padFront(); in execute_backward() local 254 const dim_t id = od * SD - padF + kd * (DD + 1); in execute_backward() 276 auto id_start = od * SD - padF; in execute_backward() 279 auto id_end = od * SD - padF + (KD - 1) * DD + KD; in execute_backward() 301 const dim_t id = od * SD - padF + kd * (DD + 1); in execute_backward() [all …]
|
H A D | nhwc_pooling.cpp | 184 const dim_t padF = pd()->padFront(); in execute_forward() local 237 const dim_t id = od * SD - padF + kd; in execute_forward() 270 const auto id_end = min(od * SD - padF + KD, ID); in execute_forward() 351 const dim_t padF = pd()->padFront(); in execute_forward() local 517 const dim_t padF = pd()->padFront(); in execute_backward() local 553 dim_t od_right = min((id + padF) / SD + 1, OD); in execute_backward() 560 const dim_t kd = id - od * SD + padF; in execute_backward() 599 auto id_start = apply_offset(od * SD, padF); in execute_backward() 602 auto id_end = min(od * SD - padF + KD, ID); in execute_backward() 661 const dim_t padF = pd()->padFront(); in execute_backward() local [all …]
|
H A D | nchw_pooling.cpp | 64 const dim_t padF = pd()->padFront(); in execute_forward() local 94 const dim_t id = od * SD - padF + kd; in execute_forward() 213 const dim_t padF = pd()->padFront(); in execute_forward() local 248 const dim_t id = od * SD - padF + kd; in execute_forward() 270 auto id_start = apply_offset(od * SD, padF); in execute_forward() 273 auto id_end = min(od * SD - padF + KD, ID); in execute_forward() 367 const dim_t padF = pd()->padFront(); in execute_backward() local 402 const dim_t id = od * SD - padF + kd; in execute_backward() 522 const dim_t padF = pd()->padFront(); in execute_backward() local 560 const dim_t id = od * SD - padF + kd; in execute_backward() [all …]
|
/dports/misc/mnn/MNN-1.2.0/backupcode/cpubackend/ |
H A D | CPULRN.cpp | 167 int halfLocalSize = mLocalSize / 2, padF = halfLocalSize + 1, padB = mLocalSize - padF; in executeWithInChannels() local 185 float* sqrtChannel = mSquare.host<float>() + tId * padSize + padF * padWidth + padF; in executeWithInChannels() 205 sqrtChannel = mSquare.host<float>() + tId * padSize + padF * padWidth + padF; in executeWithInChannels()
|