Home
last modified time | relevance | path

Searched refs:offset_w (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DPaddingKernel.cpp157 int64_t offset_w = p.offsets[ndim - 1]; in cpu_padding() local
164 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding()
169 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding()
174 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding()
187 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding()
264 int64_t offset_w = p.offsets[ndim - 1]; in cpu_padding_channels_last() local
274 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding_channels_last()
292 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding_channels_last()
338 int64_t offset_w = p.offsets[ndim - 1]; in cpu_padding_backward() local
345 int64_t iw = PaddingType::index(ow, input_width, pad_w, offset_w); in cpu_padding_backward()
[all …]
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNEReverseKernel.cpp123 … const int offset_w = (axis_bit & 0x8) ? output->info()->dimension(3) - id[3] - 1 : id[3]; in run_reverse() local
125 …reinterpret_cast<T *>(output->ptr_to_element(Coordinates(offset_x, offset_y, offset_z, offset_w))); in run_reverse()
137 … const int offset_w = (axis_bit & 0x8) ? output->info()->dimension(3) - id[3] - 1 : id[3]; in run_reverse() local
139 …erpret_cast<T *>(output->ptr_to_element(Coordinates(offset_x, offset_y, offset_z, offset_w))) = in; in run_reverse()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthtospace_op.cc163 const int offset_w = (w % block_size); in operator ()() local
165 (offset_h * block_size + offset_w) * output_depth; in operator ()()
H A Dspacetodepth_op.cc180 const int offset_w = (w % block_size); in operator ()() local
181 const int offset_d = (offset_h * block_size + offset_w) * input_depth; in operator ()()
H A Ddepthtospace_op_gpu.cu.cc52 const int offset_w = w % block_size; in D2S_NHWC() local
53 const int offset_d = (offset_h * block_size + offset_w) * output_depth; in D2S_NHWC()
H A Dspacetodepth_op_gpu.cu.cc51 const int offset_w = w % block_size; in S2D_NHWC() local
52 const int offset_d = (offset_h * block_size + offset_w) * input_depth; in S2D_NHWC()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DConvolutionLayer.cpp91 …const int offset_w = (ofm + group * (depth_out / num_groups)) * width_weights * height_weights *… in convolution_layer_nchw() local
99 … offset_in, offset_w, offset_b, offset_out, in convolution_layer_nchw()