Home
last modified time | relevance | path

Searched refs:padF (Results 1 – 9 of 9) sorted by relevance

/dports/misc/py-mxnet/incubator-mxnet-1.9.0/src/operator/
H A Dpad.cu342 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 Dpad.cu342 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 Dref_pooling.cpp75 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 Dnhwc_pooling.cpp184 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 Dnchw_pooling.cpp64 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 Dref_pooling.cpp79 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 Dnhwc_pooling.cpp184 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 Dnchw_pooling.cpp64 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 DCPULRN.cpp167 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()