Home
last modified time | relevance | path

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

/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
134 const int input_cols = 9; in TEST() local
182 const int input_cols = 9; in TEST() local
234 const int input_cols = 9; in TEST() local
289 const int input_cols = 9; in TEST() local
342 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid()
401 int input_cols = 5; in TEST() local
418 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
478 int input_cols = 5; in TEST() local
[all …]
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 …]
Ddilation_ops_gpu.cu.cc40 const T* __restrict__ filter_ptr, int batch, int input_rows, int input_cols, in DilationKernel()
80 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropInputKernel()
130 int batch, int input_rows, int input_cols, int depth, int filter_rows, 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_backward_cuboid_convolutions_test.cc34 const int input_cols = 4; in TEST() local
97 const int input_cols = 4; in TEST() local
160 const int input_cols = 4; in TEST() local
231 const int input_cols = 4; in TEST() local
303 const int input_cols = 4; in TEST() local
370 const int input_cols = 4; in TEST() local
436 const int input_cols = 4; in TEST() local
502 const int input_cols = 4; in TEST() local
569 const int input_cols = 9; in TEST() local
636 const int input_cols = 9; in TEST() local
[all …]
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.h88 Eigen::Index input_cols = input_dims[2]; in SpatialConvolutionBackwardInput() local
221 Eigen::Index input_cols = input_dims[2]; in CuboidConvolutionBackwardInput() local
Dconv_ops.cc223 int input_cols, int in_depth, int filter_rows, in Run()
239 int input_cols, int in_depth, int filter_rows, in Run()
280 int input_cols, int in_depth, int filter_rows, in Run()
294 int input_cols, int in_depth, int filter_rows, in Run()
449 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.cc368 const int32 input_cols = static_cast<int32>(input_cols_raw); in Compute() local
Dconv_ops_using_gemm.cc507 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
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc26 tensorflow::int64 input_rows, tensorflow::int64 input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
48 tensorflow::int64 input_cols, tensorflow::int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc29 tensorflow::int64 input_cols, tensorflow::int64 input_channels, in __xla_cpu_runtime_EigenConvF32()
52 tensorflow::int64 input_rows, tensorflow::int64 input_cols, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc54 int64 input_cols, int64 input_channels, int64 kernel_rows, in MKLConvImpl()
156 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d_impl.h34 Eigen::Index input_rows, Eigen::Index input_cols, in EigenConvImpl()
Dir_emitter.cc905 int64 input_cols = in HandleConvolution() local
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h143 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
167 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
330 int input_planes, input_rows, input_cols; variable
/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/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.py1853 input_cols, argument
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h2880 const int input_cols = FlatSizeSkipDim(input_dims, 0); in ConvAsGemm() local