Home
last modified time | relevance | path

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

/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
137 … const int offset_w = (axis_bit & 0x8) ? output->info()->dimension(3) - id[3] - 1 : id[3]; in run_reverse() local
/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
264 int64_t offset_w = p.offsets[ndim - 1]; in cpu_padding_channels_last() local
338 int64_t offset_w = p.offsets[ndim - 1]; in cpu_padding_backward() local
426 int64_t offset_w = p.offsets[ndim - 1]; in cpu_padding_backward_channels_last() local
/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
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Ddepthtospace_op.cc163 const int offset_w = (w % block_size); in operator ()() local
H A Dspacetodepth_op.cc180 const int offset_w = (w % block_size); in operator ()() local
H A Dspacetodepth_op_gpu.cu.cc51 const int offset_w = w % block_size; in S2D_NHWC() local
H A Ddepthtospace_op_gpu.cu.cc52 const int offset_w = w % block_size; in D2S_NHWC() local