| /external/pytorch/aten/src/ATen/native/cpu/ |
| D | AdaptiveAvgPoolKernel.cpp | 43 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool2d() local
99 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool2d_channels_last() local
192 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool2d_channels_last() local
281 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool2d_backward() local
332 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool2d_backward_channels_last() local
445 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool3d() local
512 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool3d_channels_last() local
614 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool3d_channels_last() local
712 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool3d_backward() local
772 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_avg_pool3d_backward_channels_last() local
|
| D | AvgPoolKernel.cpp | 56 int64_t ih0 = oh * dH - padH; in cpu_avg_pool2d() local
140 int64_t ih0 = oh * dH - padH; in cpu_avg_pool2d_channels_last() local
259 int64_t ih0 = oh * dH - padH; in cpu_avg_pool2d_channels_last() local
379 int64_t ih0 = oh * dH - padH; in cpu_avg_pool2d_backward() local
448 int64_t ih0 = oh * dH - padH; in cpu_avg_pool2d_backward_channels_last() local
593 int64_t ih0 = oh * dH - padH; in cpu_avg_pool3d() local
686 int64_t ih0 = oh * dH - padH; in cpu_avg_pool3d_channels_last() local
814 int64_t ih0 = oh * dH - padH; in cpu_avg_pool3d_channels_last() local
943 int64_t ih0 = oh * dH - padH; in cpu_avg_pool3d_backward() local
1021 int64_t ih0 = oh * dH - padH; in cpu_avg_pool3d_backward_channels_last() local
|
| D | AdaptiveMaxPoolKernel.cpp | 47 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_max_pool2d() local
130 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_max_pool2d_channels_last() local
248 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_max_pool2d_channels_last() local
518 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_max_pool3d() local
610 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_max_pool3d_channels_last() local
736 int64_t ih0 = start_index(oh, output_height, input_height); in cpu_adaptive_max_pool3d_channels_last() local
|
| D | MaxPoolKernel.cpp | 79 int64_t ih0, int64_t ih1, in compute_internal()
154 int64_t ih0, int64_t ih1, in compute_internal()
306 int64_t ih0 = oh * dH - padH; in cpu_max_pool() local
439 int64_t ih0 = oh * dH - padH; in cpu_max_pool_channels_last() local
|
| D | UpSampleMoreKernel.cpp | 501 int64_t ih0, ih1, iw0, iw1; in cpu_upsample_linear_backward() local
544 int64_t id0, id1, ih0, ih1, iw0, iw1; in cpu_upsample_linear_backward() local
648 int64_t ih0, ih1, iw0, iw1; in cpu_upsample_linear_backward_channels_last() local
697 int64_t id0, id1, ih0, ih1, iw0, iw1; in cpu_upsample_linear_backward_channels_last() local
|
| D | UpSampleKernel.cpp | 609 int64_t ih0 = 0, ih1 = 0, iw0 = 0, iw1 = 0; in cpu_upsample_linear_channels_last() local
658 int64_t id0 = 0, id1 = 0, ih0 = 0, ih1 = 0, iw0 = 0, iw1 = 0; in cpu_upsample_linear_channels_last() local
|
| /external/executorch/kernels/portable/cpu/util/ |
| D | kernel_ops_util.h | 195 int64_t ih0 = out_y * s_H - p_H; in kernel_reduction_then_map_2d() local
|