Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
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
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
Dconvolution_transposed_thin.h113 const int kernel_x = weights.shape.w; in RearrangeWeightsData() local
Dconv_weights_converter_test_util.cc79 const int kernel_x = weights.shape.w; in ConvolutionWeightsConverterTest() local
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
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()
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()
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()
/external/XNNPACK/src/
Dim2col.c35 for (size_t kernel_x = 0; kernel_x < kernel_width; kernel_x++) { in xnn_im2col_conv2d() local
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
/external/libaom/av1/encoder/x86/
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
/external/eigen/bench/tensors/
Dtensor_benchmarks.h492 void convolution(int num_iters, int kernel_x, int kernel_y) { in convolution()
/external/libaom/av1/encoder/
Dreconinter_enc.c482 const int16_t *const kernel_x = in aom_upsampled_pred_c() local
624 const int16_t *const kernel_x = in aom_highbd_upsampled_pred_c() local
/external/ComputeLibrary/src/cpu/operators/internal/
DCpuGemmAssemblyDispatch.cpp270 for(int64_t kernel_x = 0; kernel_x < _cp.kernel_width; kernel_x++) in prepare_indirect_buffer() local