Home
last modified time | relevance | path

Searched defs:kernel_x (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c79 for (size_t kernel_x = 0; kernel_x < kernel_width; in pytorch_qnnp_indirection_init_conv3d() local
103 for (size_t kernel_x = 0; kernel_x < kernel_width; in pytorch_qnnp_indirection_init_conv3d() local
118 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in pytorch_qnnp_indirection_init_conv3d() local
276 for (size_t kernel_x = 0; kernel_x < kernel_width; in pytorch_qnnp_indirection_init_dwconv() local
301 for (size_t kernel_x = 0; kernel_x < kernel_width; in pytorch_qnnp_indirection_init_dwconv() local
320 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in pytorch_qnnp_indirection_init_dwconv() local
382 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in pytorch_qnnp_indirection_init_deconv2d() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Ddepthwise_conv.h121 const int kernel_x = weights.shape.w; in RearrangeWeightsForDWConv2D() local
150 const int kernel_x = weights.shape.w; in UploadWeightsForDWConv2D() local
189 const int kernel_x = weights.shape.w; in RearrangeWeightsForDWConv3D() local
221 const int kernel_x = weights.shape.w; in UploadWeightsForDWConv3D() local
H A Dconv_constants.h39 const int kernel_x = weights.shape.w; in RearrangeWeightsForConvConstants() local
76 const int kernel_x = weights.shape.w; in RearrangeWeightsForConvConstantsDot() local
115 const int kernel_x = weights.shape.w; in UploadWeightsForConvConstants() local
H A Dconvolution_transposed_thin.h113 const int kernel_x = weights.shape.w; in RearrangeWeightsData() local
H A Dconv_weights_converter_test_util.cc79 const int kernel_x = weights.shape.w; in ConvolutionWeightsConverterTest() local
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c58 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_indirection_init_conv2d() local
70 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_indirection_init_conv2d() local
120 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_indirection_init_deconv2d() local
177 for (size_t kernel_x = offset_x; kernel_x < kernel_width; kernel_x += stride_width) { in xnn_indirection_init_subconv2d() local
230 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_indirection_init_dwconv2d() local
243 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_indirection_init_dwconv2d() local
H A Dim2col.c35 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_im2col_conv2d() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv3d.cc28 int64_t input_channels, int64_t kernel_x, int64_t kernel_y, in __xla_cpu_runtime_EigenConv3DF32()
53 int64_t input_z, int64_t input_channels, int64_t kernel_x, int64_t kernel_y, in __xla_cpu_runtime_EigenConv3DF16()
H A Druntime_single_threaded_conv3d.cc25 int64_t input_channels, int64_t kernel_x, int64_t kernel_y, in __xla_cpu_runtime_EigenSingleThreadedConv3DF32()
48 int64_t input_z, int64_t input_channels, int64_t kernel_x, int64_t kernel_y, in __xla_cpu_runtime_EigenSingleThreadedConv3DF16()
H A Druntime_conv_impl.h33 Eigen::Index input_y, Eigen::Index input_channels, Eigen::Index kernel_x, in EigenConv2DImpl()
111 Eigen::Index kernel_x, Eigen::Index kernel_y, Eigen::Index kernel_z, in EigenConv3DImpl()
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dreconinter_enc_sse2.c131 const int16_t *const kernel_x = in aom_upsampled_pred_sse2() local
250 const int16_t *const kernel_x = in aom_highbd_upsampled_pred_sse2() local
/aosp_15_r20/external/libaom/av1/encoder/
H A Dreconinter_enc.c482 const int16_t *const kernel_x = in aom_upsampled_pred_c() local
600 const int16_t *const kernel_x = in aom_highbd_upsampled_pred_c() local
/aosp_15_r20/external/libaom/av1/encoder/arm/
H A Dreconinter_enc_neon.c230 const int16_t *const kernel_x = in aom_highbd_upsampled_pred_neon() local
/aosp_15_r20/external/eigen/bench/tensors/
H A Dtensor_benchmarks.h492 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution()
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/
H A DCpuGemmAssemblyDispatch.cpp270 for(int64_t kernel_x = 0; kernel_x < _cp.kernel_width; kernel_x++) in prepare_indirect_buffer() local