Home
last modified time | relevance | path

Searched defs:offset_h (Results 1 – 6 of 6) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPaddingKernel.cpp156 int64_t offset_h = ndim >= 2 ? p.offsets[ndim - 2] : 0; in cpu_padding() local
263 int64_t offset_h = ndim >= 2 ? p.offsets[ndim - 2] : 0; in cpu_padding_channels_last() local
337 int64_t offset_h = ndim >= 2 ? p.offsets[ndim - 2] : 0; in cpu_padding_backward() local
425 int64_t offset_h = ndim >= 2 ? p.offsets[ndim - 2] : 0; in cpu_padding_backward_channels_last() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthtospace_op.cc160 const int offset_h = (h % block_size); in operator ()() local
H A Dspacetodepth_op.cc177 const int offset_h = (h % block_size); in operator ()() local
H A Dspacetodepth_op_gpu.cu.cc49 const int offset_h = h % block_size; in S2D_NHWC() local
H A Ddepthtospace_op_gpu.cu.cc50 const int offset_h = h % block_size; in D2S_NHWC() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
H A Doptimized_ops.h1337 for (int offset_h = 0; offset_h < op_params.block_size; ++offset_h) { in DepthToSpace() local
1378 for (int offset_h = 0; offset_h < op_params.block_size; ++offset_h) { in SpaceToDepth() local