/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
H A D | depthwise_conv.h | 122 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 D | conv_constants.h | 40 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 D | convolution_transposed_thin.h | 114 const int kernel_y = weights.shape.h; in RearrangeWeightsData() local
|
H A D | conv_weights_converter_test_util.cc | 80 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 D | indirection.c | 75 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 D | runtime_conv3d.cc | 28 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 D | runtime_single_threaded_conv3d.cc | 25 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 D | runtime_conv_impl.h | 34 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 D | indirection.c | 55 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 D | im2col.c | 32 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 D | reconinter_enc_sse2.c | 133 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 D | reconinter_enc.c | 484 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 D | reconinter_enc_neon.c | 232 const int16_t *const kernel_y = in aom_highbd_upsampled_pred_neon() local
|
/aosp_15_r20/external/eigen/bench/tensors/ |
H A D | tensor_benchmarks.h | 492 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution()
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/operators/internal/ |
H A D | CpuGemmAssemblyDispatch.cpp | 268 for(int64_t kernel_y = 0; kernel_y < _cp.kernel_height; kernel_y++) in prepare_indirect_buffer() local
|