Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc27 int64_t kernel_cols, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
45 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc29 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_EigenConvF32()
50 int64_t kernel_cols, int64_t kernel_channels, int64_t kernel_filters, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc55 int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, in MKLConvImpl()
157 int64_t input_channels, int64_t kernel_rows, int64_t kernel_cols, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h36 Eigen::Index kernel_cols, Eigen::Index kernel_channels, in EigenConvImpl()
Dir_emitter.cc918 int64_t kernel_cols = in HandleConvolution() local
/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/
Dwinograd.hpp460 static const int kernel_cols = KernelCols; member in winograd::WeightTransform
497 static constexpr int kernel_cols = KernelCols; member in winograd::WinogradGEMM
/external/ComputeLibrary/src/core/NEON/kernels/convolution/depthwise/
Ddepthwise.hpp108 static constexpr int kernel_cols = KernelCols; member in depthwise::DepthwiseConvolutionBase
/external/ruy/ruy/
Dblock_map_test.cc36 int kernel_cols, int lhs_scalar_size, in MakeBlockMapTuningTest()
Dblock_map.cc172 void GetRectangularness(int rows, int cols, int kernel_rows, int kernel_cols, in GetRectangularness()
340 int kernel_cols, int lhs_scalar_size, int rhs_scalar_size, in MakeBlockMap()
Dpack_arm.cc2257 int input_xor, int kernel_cols) { in Pack8bitRowMajorForNeon()
/external/tensorflow/tensorflow/core/kernels/
Dnn_ops_test.cc839 int kernel_cols, int stride, Padding padding, in BM_AvgPool()
929 int kernel_cols, int stride, Padding padding, in BM_AvgPoolBk()
1031 int kernel_cols, int stride, Padding padding, in BM_MaxPool()
1125 int kernel_cols, int stride, Padding padding, in BM_MaxPoolBk()
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc1130 int32_t kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in AvgPoolShape() local
1573 int32_t kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in MaxPoolShapeImpl() local
1712 int32_t kernel_cols = GetTensorDim(kernel_sizes, data_format, 'W'); in MaxPoolV2Shape() local
1769 int32_t kernel_planes, kernel_rows, kernel_cols; in Pool3DShape() local
/external/tensorflow/tensorflow/compiler/tests/
Drandomized_tests.cc1113 int kernel_cols = in TEST_F() local
2317 int kernel_cols = in TEST_F() local