Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Ddepthwise_conv.h122 const int kernel_y = weights.shape.h; in RearrangeWeightsForDWConv2D() local
151 const int kernel_y = weights.shape.h; in UploadWeightsForDWConv2D() local
190 const int kernel_y = weights.shape.h; in RearrangeWeightsForDWConv3D() local
222 const int kernel_y = weights.shape.h; in UploadWeightsForDWConv3D() local
H A Dconv_constants.h40 const int kernel_y = weights.shape.h; in RearrangeWeightsForConvConstants() local
77 const int kernel_y = weights.shape.h; in RearrangeWeightsForConvConstantsDot() local
116 const int kernel_y = weights.shape.h; in UploadWeightsForConvConstants() local
H A Dconvolution_transposed_thin.h114 const int kernel_y = weights.shape.h; in RearrangeWeightsData() local
H A Dconv_weights_converter_test_util.cc80 const int kernel_y = weights.shape.h; in ConvolutionWeightsConverterTest() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A Dindirection.c75 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in pytorch_qnnp_indirection_init_conv3d() local
117 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in pytorch_qnnp_indirection_init_conv3d() local
271 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in pytorch_qnnp_indirection_init_dwconv() local
318 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in pytorch_qnnp_indirection_init_dwconv() local
378 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in pytorch_qnnp_indirection_init_deconv2d() 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.h34 Eigen::Index kernel_y, Eigen::Index kernel_channels, in EigenConv2DImpl()
111 Eigen::Index kernel_x, Eigen::Index kernel_y, Eigen::Index kernel_z, in EigenConv3DImpl()
/aosp_15_r20/external/XNNPACK/src/
H A Dindirection.c55 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_indirection_init_conv2d() local
117 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_indirection_init_deconv2d() local
172 for (size_t kernel_y = offset_y; kernel_y < kernel_height; kernel_y += stride_height) { in xnn_indirection_init_subconv2d() local
226 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_indirection_init_dwconv2d() local
H A Dim2col.c32 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_im2col_conv2d() local
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dreconinter_enc_sse2.c133 const int16_t *const kernel_y = in aom_upsampled_pred_sse2() local
252 const int16_t *const kernel_y = in aom_highbd_upsampled_pred_sse2() local
/aosp_15_r20/external/libaom/av1/encoder/
H A Dreconinter_enc.c484 const int16_t *const kernel_y = in aom_upsampled_pred_c() local
602 const int16_t *const kernel_y = in aom_highbd_upsampled_pred_c() local
/aosp_15_r20/external/libaom/av1/encoder/arm/
H A Dreconinter_enc_neon.c232 const int16_t *const kernel_y = 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.cpp268 for(int64_t kernel_y = 0; kernel_y < _cp.kernel_height; kernel_y++) in prepare_indirect_buffer() local