Home
last modified time | relevance | path

Searched defs:input_offset (Results 1 – 25 of 1128) sorted by relevance

12345678910>>...46

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/crop/generic/neon/
H A Dinteger.cpp32 …op_window(const ITensor *input, const ITensor *output, float *output_ptr, Coordinates input_offset, in u8_in_bounds_crop_window()
39 …op_window(const ITensor *input, const ITensor *output, float *output_ptr, Coordinates input_offset, in u16_in_bounds_crop_window()
46 …op_window(const ITensor *input, const ITensor *output, float *output_ptr, Coordinates input_offset, in u32_in_bounds_crop_window()
53 …op_window(const ITensor *input, const ITensor *output, float *output_ptr, Coordinates input_offset, in s8_in_bounds_crop_window()
60 …op_window(const ITensor *input, const ITensor *output, float *output_ptr, Coordinates input_offset, in s16_in_bounds_crop_window()
67 …op_window(const ITensor *input, const ITensor *output, float *output_ptr, Coordinates input_offset, in s32_in_bounds_crop_window()
/aosp_15_r20/external/XNNPACK/test/
H A Df32-dwconv.cc99 TEST(F32_DWCONV_UP4X3__WASMSIMD, input_offset) { in TEST() argument
202 TEST(F32_DWCONV_UP4X4__WASMSIMD, input_offset) { in TEST() argument
305 TEST(F32_DWCONV_UP4X9__WASMSIMD, input_offset) { in TEST() argument
408 TEST(F32_DWCONV_UP4X9__WASMSIMD_ACC2, input_offset) { in TEST() argument
511 TEST(F32_DWCONV_UP4X25__WASMSIMD, input_offset) { in TEST() argument
614 TEST(F32_DWCONV_UP8X3__WASMSIMD, input_offset) { in TEST() argument
717 TEST(F32_DWCONV_UP8X4__WASMSIMD, input_offset) { in TEST() argument
820 TEST(F32_DWCONV_UP8X9__WASMSIMD, input_offset) { in TEST() argument
923 TEST(F32_DWCONV_UP8X9__WASMSIMD_ACC2, input_offset) { in TEST() argument
1026 TEST(F32_DWCONV_UP8X25__WASMSIMD, input_offset) { in TEST() argument
[all …]
H A Du8-ibilinear.cc47 TEST(U8_IBILINEAR__SCALAR_C1, input_offset) { in TEST() argument
116 TEST(U8_IBILINEAR__SCALAR_C2, input_offset) { in TEST() argument
185 TEST(U8_IBILINEAR__SCALAR_C4, input_offset) { in TEST() argument
260 TEST(U8_IBILINEAR__NEON_C8, input_offset) { in TEST() argument
339 TEST(U8_IBILINEAR__NEON_C16, input_offset) { in TEST() argument
418 TEST(U8_IBILINEAR__SSE2_C8, input_offset) { in TEST() argument
497 TEST(U8_IBILINEAR__SSE2_C16, input_offset) { in TEST() argument
576 TEST(U8_IBILINEAR__SSE41_C8, input_offset) { in TEST() argument
655 TEST(U8_IBILINEAR__SSE41_C16, input_offset) { in TEST() argument
729 TEST(U8_IBILINEAR__WASMSIMD_DOT16X2_C8, input_offset) { in TEST() argument
[all …]
H A Ds8-ibilinear.cc47 TEST(S8_IBILINEAR__SCALAR_C1, input_offset) { in TEST() argument
116 TEST(S8_IBILINEAR__SCALAR_C2, input_offset) { in TEST() argument
185 TEST(S8_IBILINEAR__SCALAR_C4, input_offset) { in TEST() argument
260 TEST(S8_IBILINEAR__NEON_C8, input_offset) { in TEST() argument
339 TEST(S8_IBILINEAR__NEON_C16, input_offset) { in TEST() argument
418 TEST(S8_IBILINEAR__SSE2_C8, input_offset) { in TEST() argument
497 TEST(S8_IBILINEAR__SSE2_C16, input_offset) { in TEST() argument
576 TEST(S8_IBILINEAR__SSE41_C8, input_offset) { in TEST() argument
655 TEST(S8_IBILINEAR__SSE41_C16, input_offset) { in TEST() argument
729 TEST(S8_IBILINEAR__WASMSIMD_DOT16X2_C8, input_offset) { in TEST() argument
[all …]
H A Df32-ibilinear.cc47 TEST(F32_IBILINEAR__SCALAR_C1, input_offset) { in TEST() argument
116 TEST(F32_IBILINEAR__SCALAR_C2, input_offset) { in TEST() argument
185 TEST(F32_IBILINEAR__SCALAR_C4, input_offset) { in TEST() argument
260 TEST(F32_IBILINEAR__NEON_C4, input_offset) { in TEST() argument
339 TEST(F32_IBILINEAR__NEON_C8, input_offset) { in TEST() argument
418 TEST(F32_IBILINEAR__NEONFMA_C4, input_offset) { in TEST() argument
497 TEST(F32_IBILINEAR__NEONFMA_C8, input_offset) { in TEST() argument
576 TEST(F32_IBILINEAR__SSE_C4, input_offset) { in TEST() argument
655 TEST(F32_IBILINEAR__SSE_C8, input_offset) { in TEST() argument
729 TEST(F32_IBILINEAR__WASMSIMD_C4, input_offset) { in TEST() argument
[all …]
H A Df32-ibilinear-chw.cc56 TEST(F32_IBILINEAR_CHW__SCALAR_P1, input_offset) { in TEST() argument
135 TEST(F32_IBILINEAR_CHW__SCALAR_P2, input_offset) { in TEST() argument
214 TEST(F32_IBILINEAR_CHW__SCALAR_P4, input_offset) { in TEST() argument
294 TEST(F32_IBILINEAR_CHW__WASMSIMD_P4, input_offset) { in TEST() argument
375 TEST(F32_IBILINEAR_CHW__WASMSIMD_P8, input_offset) { in TEST() argument
462 TEST(F32_IBILINEAR_CHW__NEON_P4, input_offset) { in TEST() argument
551 TEST(F32_IBILINEAR_CHW__NEON_P8, input_offset) { in TEST() argument
640 TEST(F32_IBILINEAR_CHW__NEON_P16, input_offset) { in TEST() argument
729 TEST(F32_IBILINEAR_CHW__NEONFMA_P4, input_offset) { in TEST() argument
818 TEST(F32_IBILINEAR_CHW__NEONFMA_P8, input_offset) { in TEST() argument
[all …]
H A Dqc8-dwconv-minmax-fp32.cc180 TEST(QC8_DWCONV_MINMAX_FP32_UP8X3__AARCH32_NEONV8_MLA8_CORTEX_A35, input_offset) { in TEST() argument
366 TEST(QC8_DWCONV_MINMAX_FP32_UP8X3__NEON_MLA8_LD64, input_offset) { in TEST() argument
552 TEST(QC8_DWCONV_MINMAX_FP32_UP8X3__NEONV8_MLA8_LD64, input_offset) { in TEST() argument
738 TEST(QC8_DWCONV_MINMAX_FP32_UP8X9__NEON_MLA8_LD64, input_offset) { in TEST() argument
924 TEST(QC8_DWCONV_MINMAX_FP32_UP8X9__NEON_MUL8_LD64, input_offset) { in TEST() argument
1110 TEST(QC8_DWCONV_MINMAX_FP32_UP8X9__NEON_MUL16, input_offset) { in TEST() argument
1296 TEST(QC8_DWCONV_MINMAX_FP32_UP8X9__NEONV8_MLA8_LD64, input_offset) { in TEST() argument
1482 TEST(QC8_DWCONV_MINMAX_FP32_UP8X9__NEONV8_MUL8_LD64, input_offset) { in TEST() argument
1668 TEST(QC8_DWCONV_MINMAX_FP32_UP8X9__NEONV8_MUL16, input_offset) { in TEST() argument
1854 TEST(QC8_DWCONV_MINMAX_FP32_UP8X25__NEON_MLA8_LD64, input_offset) { in TEST() argument
[all …]
H A Dqs8-dwconv-minmax-fp32.cc180 TEST(QS8_DWCONV_MINMAX_FP32_UP8X9__NEON_MUL16, input_offset) { in TEST() argument
366 TEST(QS8_DWCONV_MINMAX_FP32_UP8X9__NEONV8_MUL16, input_offset) { in TEST() argument
552 TEST(QS8_DWCONV_MINMAX_FP32_UP8X25__NEON_MUL16, input_offset) { in TEST() argument
738 TEST(QS8_DWCONV_MINMAX_FP32_UP8X25__NEONV8_MUL16, input_offset) { in TEST() argument
924 TEST(QS8_DWCONV_MINMAX_FP32_UP16X9__NEON_MUL16, input_offset) { in TEST() argument
1110 TEST(QS8_DWCONV_MINMAX_FP32_UP16X9__NEONV8_MUL16, input_offset) { in TEST() argument
1296 TEST(QS8_DWCONV_MINMAX_FP32_UP16X25__NEON_MUL16, input_offset) { in TEST() argument
1482 TEST(QS8_DWCONV_MINMAX_FP32_UP16X25__NEONV8_MUL16, input_offset) { in TEST() argument
1668 TEST(QS8_DWCONV_MINMAX_FP32_UP24X9__NEON_MUL16, input_offset) { in TEST() argument
1854 TEST(QS8_DWCONV_MINMAX_FP32_UP24X9__NEONV8_MUL16, input_offset) { in TEST() argument
[all …]
H A Df32-dwconv-minmax.cc180 TEST(F32_DWCONV_MINMAX_UP4X9__AARCH64_NEONFMA, input_offset) { in TEST() argument
375 TEST(F32_DWCONV_MINMAX_UP4X9__AARCH64_NEONFMA_CORTEX_A55, input_offset) { in TEST() argument
561 TEST(F32_DWCONV_MINMAX_UP4X3__NEON, input_offset) { in TEST() argument
747 TEST(F32_DWCONV_MINMAX_UP4X3__NEON_ACC2, input_offset) { in TEST() argument
933 TEST(F32_DWCONV_MINMAX_UP4X3__NEONFMA, input_offset) { in TEST() argument
1119 TEST(F32_DWCONV_MINMAX_UP4X3__NEONFMA_ACC2, input_offset) { in TEST() argument
1305 TEST(F32_DWCONV_MINMAX_UP4X4__NEON, input_offset) { in TEST() argument
1491 TEST(F32_DWCONV_MINMAX_UP4X4__NEON_ACC2, input_offset) { in TEST() argument
1677 TEST(F32_DWCONV_MINMAX_UP4X4__NEONFMA, input_offset) { in TEST() argument
1863 TEST(F32_DWCONV_MINMAX_UP4X4__NEONFMA_ACC2, input_offset) { in TEST() argument
[all …]
H A Df16-dwconv-minmax.cc180 TEST(F16_DWCONV_MINMAX_UP8X3__NEONFP16ARITH, input_offset) { in TEST() argument
366 TEST(F16_DWCONV_MINMAX_UP8X3__NEONFP16ARITH_ACC2, input_offset) { in TEST() argument
552 TEST(F16_DWCONV_MINMAX_UP8X4__NEONFP16ARITH, input_offset) { in TEST() argument
738 TEST(F16_DWCONV_MINMAX_UP8X4__NEONFP16ARITH_ACC2, input_offset) { in TEST() argument
924 TEST(F16_DWCONV_MINMAX_UP8X9__NEONFP16ARITH, input_offset) { in TEST() argument
1110 TEST(F16_DWCONV_MINMAX_UP8X9__NEONFP16ARITH_ACC2, input_offset) { in TEST() argument
1296 TEST(F16_DWCONV_MINMAX_UP8X25__NEONFP16ARITH, input_offset) { in TEST() argument
1482 TEST(F16_DWCONV_MINMAX_UP8X25__NEONFP16ARITH_ACC2, input_offset) { in TEST() argument
1668 TEST(F16_DWCONV_MINMAX_UP16X3__NEONFP16ARITH, input_offset) { in TEST() argument
1854 TEST(F16_DWCONV_MINMAX_UP16X3__NEONFP16ARITH_ACC2, input_offset) { in TEST() argument
[all …]
H A Dqu8-dwconv-minmax-fp32.cc208 TEST(QU8_DWCONV_MINMAX_FP32_UP8X9__NEON_MUL16, input_offset) { in TEST() argument
422 TEST(QU8_DWCONV_MINMAX_FP32_UP8X9__NEONV8_MUL16, input_offset) { in TEST() argument
636 TEST(QU8_DWCONV_MINMAX_FP32_UP8X25__NEON_MUL16, input_offset) { in TEST() argument
850 TEST(QU8_DWCONV_MINMAX_FP32_UP8X25__NEONV8_MUL16, input_offset) { in TEST() argument
1064 TEST(QU8_DWCONV_MINMAX_FP32_UP16X9__NEON_MUL16, input_offset) { in TEST() argument
1278 TEST(QU8_DWCONV_MINMAX_FP32_UP16X9__NEONV8_MUL16, input_offset) { in TEST() argument
1492 TEST(QU8_DWCONV_MINMAX_FP32_UP16X25__NEON_MUL16, input_offset) { in TEST() argument
1706 TEST(QU8_DWCONV_MINMAX_FP32_UP16X25__NEONV8_MUL16, input_offset) { in TEST() argument
1920 TEST(QU8_DWCONV_MINMAX_FP32_UP24X9__NEON_MUL16, input_offset) { in TEST() argument
2134 TEST(QU8_DWCONV_MINMAX_FP32_UP24X9__NEONV8_MUL16, input_offset) { in TEST() argument
[all …]
H A Dqs8-dwconv-minmax-rndnu.cc180 TEST(QS8_DWCONV_MINMAX_RNDNU_UP8X9__NEON_MLA8_LD64, input_offset) { in TEST() argument
366 TEST(QS8_DWCONV_MINMAX_RNDNU_UP8X9__NEON_MUL8_LD64, input_offset) { in TEST() argument
552 TEST(QS8_DWCONV_MINMAX_RNDNU_UP8X9__NEON_MUL16, input_offset) { in TEST() argument
738 TEST(QS8_DWCONV_MINMAX_RNDNU_UP8X25__NEON_MLA8_LD64, input_offset) { in TEST() argument
924 TEST(QS8_DWCONV_MINMAX_RNDNU_UP8X25__NEON_MUL8_LD64, input_offset) { in TEST() argument
1110 TEST(QS8_DWCONV_MINMAX_RNDNU_UP8X25__NEON_MUL16, input_offset) { in TEST() argument
1296 TEST(QS8_DWCONV_MINMAX_RNDNU_UP16X9__NEON_MLA8_LD64, input_offset) { in TEST() argument
1482 TEST(QS8_DWCONV_MINMAX_RNDNU_UP16X9__NEON_MLA8_LD128, input_offset) { in TEST() argument
1668 TEST(QS8_DWCONV_MINMAX_RNDNU_UP16X9__NEON_MUL8_LD64, input_offset) { in TEST() argument
1854 TEST(QS8_DWCONV_MINMAX_RNDNU_UP16X9__NEON_MUL8_LD128, input_offset) { in TEST() argument
[all …]
H A Df16-ibilinear.cc71 TEST(F16_IBILINEAR__NEONFP16ARITH_C8, input_offset) { in TEST() argument
150 TEST(F16_IBILINEAR__NEONFP16ARITH_C16, input_offset) { in TEST() argument
229 TEST(F16_IBILINEAR__FMA3_C8, input_offset) { in TEST() argument
308 TEST(F16_IBILINEAR__FMA3_C16, input_offset) { in TEST() argument
H A Dqu8-dwconv-minmax-rndnu.cc208 TEST(QU8_DWCONV_MINMAX_RNDNU_UP8X9__NEON_MUL8, input_offset) { in TEST() argument
422 TEST(QU8_DWCONV_MINMAX_RNDNU_UP8X9__NEON_MUL16, input_offset) { in TEST() argument
636 TEST(QU8_DWCONV_MINMAX_RNDNU_UP8X25__NEON_MUL8, input_offset) { in TEST() argument
850 TEST(QU8_DWCONV_MINMAX_RNDNU_UP8X25__NEON_MUL16, input_offset) { in TEST() argument
1064 TEST(QU8_DWCONV_MINMAX_RNDNU_UP16X9__NEON_MUL8, input_offset) { in TEST() argument
1278 TEST(QU8_DWCONV_MINMAX_RNDNU_UP16X9__NEON_MUL16, input_offset) { in TEST() argument
1492 TEST(QU8_DWCONV_MINMAX_RNDNU_UP16X25__NEON_MUL8, input_offset) { in TEST() argument
1706 TEST(QU8_DWCONV_MINMAX_RNDNU_UP16X25__NEON_MUL16, input_offset) { in TEST() argument
1920 TEST(QU8_DWCONV_MINMAX_RNDNU_UP24X9__NEON_MUL8, input_offset) { in TEST() argument
2134 TEST(QU8_DWCONV_MINMAX_RNDNU_UP24X9__NEON_MUL16, input_offset) { in TEST() argument
[all …]
H A Dargmaxpool-microkernel-tester.h50 inline ArgMaxPoolMicrokernelTester& input_offset(size_t input_offset) { in input_offset() argument
56 inline size_t input_offset() const { in input_offset() function
H A Df16-ibilinear-chw.cc81 TEST(F16_IBILINEAR_CHW__NEONFP16ARITH_P4, input_offset) { in TEST() argument
170 TEST(F16_IBILINEAR_CHW__NEONFP16ARITH_P8, input_offset) { in TEST() argument
259 TEST(F16_IBILINEAR_CHW__NEONFP16ARITH_P16, input_offset) { in TEST() argument
H A Dibilinear-microkernel-tester.h48 inline IBilinearMicrokernelTester& input_offset(uint32_t input_offset) { in input_offset() function
53 inline uint32_t input_offset() const { in input_offset() function
H A Dmaxpool-microkernel-tester.h50 inline MaxPoolMicrokernelTester& input_offset(size_t input_offset) { in input_offset() function
56 inline size_t input_offset() const { in input_offset() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/
H A Dlogsoftmax_quantized_test.cc39 int32 input_offset, const double input_scale, in RunLogSoftmaxFloatReference()
72 int32 input_offset, const double input_scale, in RunLogSoftmaxFloatReference()
147 const RuntimeShape& shape_common, int32 input_offset, in RunOneLogSoftmaxTest()
205 const RuntimeShape& shape_common, int32 input_offset, in RunOneLogSoftmaxTest()
261 const int32 input_offset = UniformRandomInt(-256, 0); in TryOneUniformLogSoftmax() local
294 const int32 input_offset = UniformRandomInt(-256, 0); in TryOneSkyscraperLogSoftmax() local
H A Dsoftmax_quantized_test.cc37 int32 input_offset, const double input_scale, in RunSoftmaxFloatReference()
107 const RuntimeShape& shape_common, int32 input_offset, in RunOneSoftmaxTest()
183 const int32 input_offset = UniformRandomInt(-256, 0); in TryOneUniformSoftmax() local
216 const int32 input_offset = UniformRandomInt(-256, 0); in TryOneSkyscraperSoftmax() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUpSampleMoreKernel.cpp133 int64_t input_offset = buffer_data.get() == nullptr ? c * input_slice_size : 0; in cpu_upsample_nearest_backward() local
158 int64_t input_offset = buffer_data.get() == nullptr ? c * input_slice_size : 0; in cpu_upsample_nearest_backward() local
186 int64_t input_offset = buffer_data.get() == nullptr ? c * input_slice_size : 0; in cpu_upsample_nearest_backward() local
268 int64_t input_offset = buffer_data.get() == nullptr ? n * input_slice_size : 0; in cpu_upsample_nearest_backward_channels_last() local
299 int64_t input_offset = buffer_data.get() == nullptr ? n * input_slice_size : 0; in cpu_upsample_nearest_backward_channels_last() local
469 int64_t input_offset = buffer_data.get() == nullptr ? c * input_slice_size : 0; in cpu_upsample_linear_backward() local
504 int64_t input_offset = buffer_data.get() == nullptr ? c * input_slice_size : 0; in cpu_upsample_linear_backward() local
547 int64_t input_offset = buffer_data.get() == nullptr ? c * input_slice_size : 0; in cpu_upsample_linear_backward() local
651 int64_t input_offset = buffer_data.get() == nullptr ? n * input_slice_size : 0; in cpu_upsample_linear_backward_channels_last() local
700 int64_t input_offset = buffer_data.get() == nullptr ? n * input_slice_size : 0; in cpu_upsample_linear_backward_channels_last() local
H A DPixelShuffleKernel.cpp46 int64_t input_offset = n * stride_n + c * stride_c + s1 * stride_s1 + in cpu_pixel_shuffle() local
145 int64_t input_offset = n * stride_n + c * stride_c + h * stride_h + in cpu_pixel_unshuffle() local
187 int64_t input_offset = n * stride_n + h * stride_h + s1 * stride_s1 + in cpu_pixel_unshuffle_channels_last() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/detail/
H A DNEDirectConvolutionDetail.h250 const size_t dilation_x, int input_offset) in single_convolve_3x3_dilation()
354 unsigned int stridex, int input_offset) in convolve_3x3()
468 size_t dilation_x, int32_t input_offset) in single_convolve_3x3_dilation()
555 … const size_t dilation_x, unsigned int stridex, int input_offset) in convolve_3x3_dilation()
595 unsigned int stridex, int32_t input_offset) in convolve_3x3()
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/depthwiseconv2d/generic/neon/
H A Dimpl.cpp181 int64_t input_offset = base_input_offset; in depthwise_loop_multiplier1_fp() local
215 int64_t input_offset = base_input_offset; in depthwise_loop_multiplier1_fp() local
285 … int input_offset = input_y * run_info.input_stride_y + input_z * run_info.input_stride_z; in depthwise_loop_generic_fp() local
388 auto input_offset = base_input_offset; in depthwise_loop_multiplier1_quantized() local
451 auto input_offset = base_input_offset; in depthwise_loop_multiplier1_quantized() local
554 …int64_t input_offset = input_y * run_info.input_stride_y + input_z * run_info.input_stride_z; in depthwise_loop_generic_quantized() local
668 …int64_t input_offset = input_y * run_info.input_stride_y + input_z * run_info.input_stride_z; in depthwise_loop_pow2_quantized_per_tensor() local
/aosp_15_r20/external/XNNPACK/src/
H A Doperator-run.c647 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_dwconv_unipass() local
682 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_argmax_pooling_unipass() local
701 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_argmax_pooling_multipass() local
723 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_max_pooling() local
761 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_average_pooling_unipass() local
779 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_average_pooling_multipass() local
800 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_pixelwise_average_pooling_unipass() local
820 const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride; in xnn_compute_pixelwise_average_pooling_multipass() local
923 …const size_t input_offset = context->input_offset + batch_index * context->input_batch_stride + ch… in xnn_compute_resize_bilinear_chw() local

12345678910>>...46