Home
last modified time | relevance | path

Searched defs:offset_d (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPaddingKernel.cpp155 int64_t offset_d = ndim == 3 ? p.offsets[ndim - 3] : 0; in cpu_padding() local
262 int64_t offset_d = ndim == 3 ? p.offsets[ndim - 3] : 0; in cpu_padding_channels_last() local
336 int64_t offset_d = ndim == 3 ? p.offsets[ndim - 3] : 0; in cpu_padding_backward() local
424 int64_t offset_d = ndim == 3 ? p.offsets[ndim - 3] : 0; in cpu_padding_backward_channels_last() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthtospace_op.cc164 const int offset_d = in operator ()() local
H A Dspacetodepth_op.cc181 const int offset_d = (offset_h * block_size + offset_w) * input_depth; in operator ()() local
H A Dspacetodepth_op_gpu.cu.cc52 const int offset_d = (offset_h * block_size + offset_w) * input_depth; in S2D_NHWC() local
H A Ddepthtospace_op_gpu.cu.cc53 const int offset_d = (offset_h * block_size + offset_w) * output_depth; in D2S_NHWC() local