Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Ddepthwise_conv.h42 const int kernel_y = weights.shape.h; in RearrangeWeightsForDWConv2D() local
46 for (int y = 0; y < kernel_y; ++y) { in RearrangeWeightsForDWConv2D()
73 const int kernel_y = weights.shape.h; in UploadWeightsForDWConv2D() local
75 const int elements_count = kernel_x * kernel_y * dst_slices; in UploadWeightsForDWConv2D()
100 desc.size = int2(kernel_x * kernel_y, dst_slices); in UploadWeightsForDWConv2D()
113 const int kernel_y = weights.shape.h; in RearrangeWeightsForDWConv3D() local
119 for (int y = 0; y < kernel_y; ++y) { in RearrangeWeightsForDWConv3D()
147 const int kernel_y = weights.shape.h; in UploadWeightsForDWConv3D() local
150 const int elements_count = kernel_x * kernel_y * kernel_z * dst_slices; in UploadWeightsForDWConv3D()
176 desc.size = int2(kernel_x * kernel_y * kernel_z, dst_slices); in UploadWeightsForDWConv3D()
Dconv_constants.h39 const int kernel_y = weights.shape.h; in RearrangeWeightsForConvConstants() local
43 for (int y = 0; y < kernel_y; ++y) { in RearrangeWeightsForConvConstants()
76 const int kernel_y = weights.shape.h; in RearrangeWeightsForConvConstantsDot() local
80 for (int y = 0; y < kernel_y; ++y) { in RearrangeWeightsForConvConstantsDot()
114 const int kernel_y = weights.shape.h; in UploadWeightsForConvConstants() local
120 const int float_count = aligned_ch_count * kernel_x * kernel_y; in UploadWeightsForConvConstants()
Dconvolution_transposed.h123 const int kernel_y = weights.shape.h; in UploadWeights() local
125 int texture_height = src_depth * kernel_x * kernel_y; in UploadWeights()
147 const int kernel_y = weights.shape.h; in UploadWeights() local
151 kernel_x * kernel_y * kernel_z * src_depth * dst_depth * 4; in UploadWeights()
187 int texture_height = src_depth * kernel_x * kernel_y * kernel_z; in UploadWeights()
Dconvolution_transposed_thin.h111 const int kernel_y = weights.shape.h; in RearrangeWeightsData() local
115 for (int y = 0; y < kernel_y; ++y) { in RearrangeWeightsData()
Dconv_weights_converter_test_util.cc67 const int kernel_y = weights.shape.h; in ConvolutionWeightsConverterTest() local
69 int texture_height = src_depth * kernel_x * kernel_y; in ConvolutionWeightsConverterTest()
Dconv_powervr.h281 const int kernel_y = weights.shape.h; in UploadWeights() local
283 int texture_height = src_depth * kernel_x * kernel_y; in UploadWeights()
/external/XNNPACK/src/
Dindirection.c52 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_indirection_init_conv2d() local
53 … const size_t input_y = output_y * stride_height + kernel_y * dilation_height - input_padding_top; in xnn_indirection_init_conv2d()
57 const size_t kernel_index = kernel_y * kernel_width + kernel_x; in xnn_indirection_init_conv2d()
68 const size_t kernel_index = kernel_y * kernel_width + kernel_x; in xnn_indirection_init_conv2d()
114 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_indirection_init_deconv2d() local
115 const size_t y = output_y + padding_top - kernel_y * dilation_height; in xnn_indirection_init_deconv2d()
120 const size_t kernel_index = kernel_y * kernel_width + kernel_x; in xnn_indirection_init_deconv2d()
169 for (size_t kernel_y = offset_y; kernel_y < kernel_height; kernel_y += stride_height) { in xnn_indirection_init_subconv2d() local
170 assert(doz(output_y + padding_top, kernel_y) % stride_height == 0); in xnn_indirection_init_subconv2d()
171 const size_t y = output_y + padding_top - kernel_y; in xnn_indirection_init_subconv2d()
[all …]
Dim2col.c32 for (size_t kernel_y = 0; kernel_y < kernel_height; kernel_y++) { in xnn_im2col_conv2d() local
33 …const size_t input_y = output_y * subsampling_height + kernel_y * dilation_height - input_padding_… in xnn_im2col_conv2d()
/external/eigen/bench/tensors/
Dtensor_benchmarks.h417 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution() argument
424 kernel_sizes[1] = kernel_y; in convolution()
428 result_sizes[1] = n_ - kernel_y + 1; in convolution()
441 (m_ - kernel_x + 1) * (n_ - kernel_y + 1) * kernel_x * kernel_y * num_iters); in convolution()
/external/tensorflow/tensorflow/python/keras/layers/
Dlocal_test.py298 kernel_y = () if width == 1 else (2,)
306 'kernel_size': kernel_x + kernel_y,
/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
Doperation_selector.cc357 const int kernel_y = weights_shape.h; in GPUOperationFromNode() local
365 int texture_height = src_depth * kernel_x * kernel_y; in GPUOperationFromNode()
/external/libaom/libaom/aom_dsp/
Dvariance.c344 const int16_t *const kernel_y = in aom_upsampled_pred_c() local
353 MAX_SB_SIZE, comp_pred, width, NULL, -1, kernel_y, 16, in aom_upsampled_pred_c()
889 const int16_t *const kernel_y = in aom_highbd_upsampled_pred_c() local
900 MAX_SB_SIZE, comp_pred8, width, NULL, -1, kernel_y, 16, width, height, in aom_highbd_upsampled_pred_c()
/external/libaom/libaom/aom_dsp/x86/
Dvariance_sse2.c596 const int16_t *const kernel_y = in aom_upsampled_pred_sse2() local
609 kernel_y, 16, width, height); in aom_upsampled_pred_sse2()
Dhighbd_variance_sse2.c702 const int16_t *const kernel_y = in aom_highbd_upsampled_pred_sse2() local
716 comp_pred8, width, NULL, -1, kernel_y, 16, width, in aom_highbd_upsampled_pred_sse2()
/external/llvm-project/llvm/test/CodeGen/Thumb2/
Dmve-gather-scatter-optimisation.ll842 …t %output_ch, i16 zeroext %ch_mult, i16 zeroext %kernel_x, i16 zeroext %kernel_y, i16 zeroext %pad…