Home
last modified time | relevance | path

Searched defs:input_cols (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc33 const int input_cols = 4; in TEST() local
82 const int input_cols = 4; in TEST() local
135 const int input_cols = 4; in TEST() local
198 const int input_cols = 4; in TEST() local
260 const int input_cols = 9; in TEST() local
308 const int input_cols = 9; in TEST() local
360 const int input_cols = 4; in TEST() local
431 const int input_cols = 4; in TEST() local
502 const int input_cols = 9; in TEST() local
557 const int input_cols = 9; in TEST() local
[all …]
Ddilation_ops_gpu.cu.cc41 int input_cols, int depth, int filter_rows, in DilationKernel()
81 const T* out_backprop_ptr, int batch, int input_rows, int input_cols, in DilationBackpropInputKernel()
130 const T* out_backprop_ptr, int batch, int input_rows, int input_cols, in DilationBackpropFilterKernel()
187 const int input_cols = input.dimension(2); in operator ()() local
218 const int input_cols = input.dimension(2); in operator ()() local
260 const int input_cols = input.dimension(2); in operator ()() local
Deigen_pooling_test.cc30 const int input_cols = 5; in TEST() local
78 const int input_cols = 5; in TEST() local
128 const int input_cols = 5; in TEST() local
188 const int input_cols = 5; in TEST() local
248 const int input_cols = 5; in TEST() local
310 const int input_cols = 5; in TEST() local
372 const int input_cols = 5; in TEST() local
450 const int input_cols = 5; in TEST() local
527 const int input_cols = 5; in TEST() local
575 const int input_cols = 5; in TEST() local
[all …]
Dmkl_conv_ops.h137 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
161 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
324 int input_planes, input_rows, input_cols; variable
Ddilation_ops.cc76 const int input_cols = input.dim_size(2); in ParseSizes() local
170 const int input_cols = input.dimension(2); in operator ()() local
279 const int input_cols = input.dimension(2); in operator ()() local
399 const int input_cols = input.dimension(2); in operator ()() local
Deigen_benchmark.h89 Eigen::Index input_cols = input_dims[2]; in SpatialConvolutionBackwardInput() local
226 Eigen::Index input_cols = input_dims[2]; in CuboidConvolutionBackwardInput() local
Dconv_ops.cc156 int input_cols, int in_depth, int filter_rows, in Run()
172 int input_cols, int in_depth, int filter_rows, in Run()
213 int input_cols, int in_depth, int filter_rows, in Run()
227 int input_cols, int in_depth, int filter_rows, in Run()
382 const int input_cols = static_cast<int>(input_cols_raw); in ComputeConv2DDimension() local
Deigen_spatial_convolutions_test.cc32 const int input_cols = 5; in TEST() local
77 const int input_cols = 5; in TEST() local
196 const int input_cols = 5; in TEST() local
248 const int input_cols = 5; in TEST() local
306 const int input_cols = 5; in TEST() local
362 const int input_cols = 5; in TEST() local
410 const int input_cols = 3; in TEST() local
458 const int input_cols = 5; in TEST() local
509 const int input_cols = 7; in TEST() local
560 const int input_cols = 7; in TEST() local
[all …]
Ddepthwise_conv_op.cc342 const int32 input_cols = static_cast<int32>(input_cols_raw); in Compute() local
Dconv_ops_using_gemm.cc505 const int input_cols = static_cast<int>(input_cols_raw); in Compute() local
Dquantized_conv_ops.cc529 const int64 input_cols = input.dim_size(2); in Compute() local
Dmkl_conv_ops.cc507 const int input_cols = static_cast<int>(input_cols_raw); in Compute() local
/external/libjpeg-turbo/simd/loongson/
Djcsample.h11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
/external/libjpeg-turbo/simd/powerpc/
Djcsample.h11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc27 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc29 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenConvF32()
47 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc54 int64 input_cols, int64 input_channels, int64 kernel_rows, in MKLConvImpl()
154 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h30 int64 input_cols, int64 input_channels, int64 kernel_rows, in EigenConvImpl()
Dir_emitter.cc1128 int64 input_cols = in HandleConvolution() local
/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch.cpp187 int input_cols = 3; in test_patch_padding_valid() local
263 int input_cols = 5; in test_patch_padding_valid_same_value() local
336 int input_cols = 2; in test_patch_padding_same() local
/external/tensorflow/tensorflow/core/kernels/neon/
Dneon_depthwise_conv_op.cc80 const int32 input_cols = input.dim_size(2); in Compute() local
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
Dfactorization_ops.py189 input_cols, argument
/external/libjpeg-turbo/
Djcsample.c94 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1638 def ConstructAndTestGradient(self, batch, input_rows, input_cols, filter_rows, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h755 const int input_cols = FlatSizeSkipDim(input_dims, 0); in ConvAsGemm() local

12