| /external/tensorflow/tensorflow/core/kernels/ |
| D | eigen_backward_spatial_convolutions_test.cc | 33 const int input_cols = 4; in TEST() local
38 const int output_cols = input_cols - patch_cols + 1; in TEST()
40 Tensor<float, 3> input_backward(input_depth, input_rows, input_cols); in TEST()
49 input_rows, input_cols, 1); in TEST()
53 EXPECT_EQ(input_backward.dimension(2), input_cols); in TEST()
57 for (int j = 0; j < input_cols; ++j) { in TEST()
82 const int input_cols = 4; in TEST() local
87 const int output_cols = input_cols - patch_cols + 1; in TEST()
89 Tensor<float, 3, RowMajor> input_backward(input_cols, input_rows, in TEST()
101 input_rows, input_cols, 1); in TEST()
[all …]
|
| D | eigen_backward_cuboid_convolutions_test.cc | 34 const int input_cols = 4; in TEST() local
39 const int output_cols = input_cols - patch_cols + 1; in TEST()
44 input_cols); in TEST()
55 kernel, output_backward, input_planes, input_rows, input_cols); in TEST()
57 EXPECT_EQ(input_backward.dimension(3), input_cols); in TEST()
65 for (int k = 0; k < input_cols; ++k) { in TEST()
97 const int input_cols = 4; in TEST() local
102 const int output_cols = input_cols - patch_cols + 1; in TEST()
106 Tensor<float, 4, RowMajor> input_backward(input_cols, input_rows, in TEST()
118 kernel, output_backward, input_planes, input_rows, input_cols); in TEST()
[all …]
|
| D | dilation_ops_gpu.cu.cc | 40 const T* __restrict__ filter_ptr, int batch, int input_rows, int input_cols, in DilationKernel() argument
60 if (w_in >= 0 && w_in < input_cols) { in DilationKernel()
63 input_cols * (h_in + input_rows * b))] + in DilationKernel()
80 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropInputKernel() argument
105 if (w_in >= 0 && w_in < input_cols) { in DilationBackpropInputKernel()
108 input_cols * (h_in + input_rows * b))] + in DilationBackpropInputKernel()
121 depth * (w_in_max + input_cols * (h_in_max + input_rows * b)), in DilationBackpropInputKernel()
130 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropFilterKernel() argument
155 if (w_in >= 0 && w_in < input_cols) { in DilationBackpropFilterKernel()
158 input_cols * (h_in + input_rows * b))] + in DilationBackpropFilterKernel()
[all …]
|
| D | eigen_pooling_test.cc | 30 const int input_cols = 5; in TEST() local
37 Tensor<float, 4> input(depth, input_rows, input_cols, num_batches); in TEST()
78 const int input_cols = 5; in TEST() local
85 Tensor<float, 4, RowMajor> input(num_batches, input_cols, input_rows, depth); in TEST()
128 const int input_cols = 5; in TEST() local
137 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols, in TEST()
188 const int input_cols = 5; in TEST() local
197 Tensor<float, 5, RowMajor> input(num_batches, input_cols, input_rows, in TEST()
248 const int input_cols = 5; in TEST() local
257 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols, in TEST()
[all …]
|
| D | eigen_spatial_convolutions_test.cc | 32 const int input_cols = 5; in TEST() local
37 const int output_cols = input_cols; in TEST()
39 Tensor<float, 3> input(input_depth, input_rows, input_cols); in TEST()
77 const int input_cols = 5; in TEST() local
82 const int output_cols = input_cols; in TEST()
84 Tensor<float, 3, RowMajor> input(input_cols, input_rows, input_depth); in TEST()
196 const int input_cols = 5; in TEST() local
202 const int output_cols = input_cols - patch_cols + 1; in TEST()
204 Tensor<float, 4> input(input_depth, input_rows, input_cols, num_batches); in TEST()
248 const int input_cols = 5; in TEST() local
[all …]
|
| D | dilation_ops.cc | 76 const int input_cols = input.dim_size(2); in ParseSizes() local
110 context, GetWindowedOutputSize(input_cols, filter_cols_eff, *stride_cols, in ParseSizes()
170 const int input_cols = input.dimension(2); in operator ()() local
193 if (w_in >= 0 && w_in < input_cols) { in operator ()()
279 const int input_cols = input.dimension(2); in operator ()() local
310 if (w_in >= 0 && w_in < input_cols) { in operator ()()
399 const int input_cols = input.dimension(2); in operator ()() local
430 if (w_in >= 0 && w_in < input_cols) { in operator ()()
|
| D | conv_ops.cc | 223 int input_cols, int in_depth, int filter_rows, in Run() argument
239 int input_cols, int in_depth, int filter_rows, in Run() argument
254 args.in_cols = input_cols; in Run()
280 int input_cols, int in_depth, int filter_rows, in Run() argument
294 int input_cols, int in_depth, int filter_rows, in Run() argument
306 desc.W = input_cols; in Run()
449 const int input_cols = static_cast<int>(input_cols_raw); in ComputeConv2DDimension() local
481 input_cols, filter_cols, dilation_cols, stride_cols, params.padding, in ComputeConv2DDimension()
486 dimensions->input_cols = input_cols; in ComputeConv2DDimension()
542 << ", input_cols = " << dimensions.input_cols in Compute()
[all …]
|
| D | eigen_benchmark.h | 88 Eigen::Index input_cols = input_dims[2]; in SpatialConvolutionBackwardInput() local
106 filter, output_backward, input_rows, input_cols); in SpatialConvolutionBackwardInput()
221 Eigen::Index input_cols = input_dims[2]; in CuboidConvolutionBackwardInput() local
240 filter, output_backward, input_planes, input_rows, input_cols); in CuboidConvolutionBackwardInput()
|
| D | conv_ops_using_gemm.cc | 507 const int input_cols = static_cast<int>(input_cols_raw); in Compute() local
527 GetWindowedOutputSize(input_cols, filter_cols, stride_cols, in Compute()
538 << ", input_cols = " << input_cols in Compute()
551 conv_functor(context, input.flat<T>().data(), batch, input_rows, input_cols, in Compute()
|
| D | depthwise_conv_op.cc | 368 const int32 input_cols = static_cast<int32>(input_cols_raw); in Compute() local
386 input_cols, filter_cols, stride_, padding_, in Compute()
417 << " Input: [" << batch << ", " << input_rows << ", " << input_cols in Compute()
454 args.in_cols = input_cols; in Compute()
|
| D | conv_ops.h | 84 int input_cols;
|
| D | quantized_conv_ops.cc | 529 const int64 input_cols = input.dim_size(2); in Compute() local
545 GetWindowedOutputSize(input_cols, filter_cols, stride, in Compute()
562 input_cols, in_depth, offset_input, filter.flat<T2>().data(), in Compute()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | runtime_single_threaded_conv2d.cc | 26 tensorflow::int64 input_rows, tensorflow::int64 input_cols, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
38 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
48 tensorflow::int64 input_cols, tensorflow::int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
59 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
|
| D | runtime_conv2d.cc | 29 tensorflow::int64 input_cols, tensorflow::int64 input_channels, in __xla_cpu_runtime_EigenConvF32() argument
43 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF32()
52 tensorflow::int64 input_rows, tensorflow::int64 input_cols, in __xla_cpu_runtime_EigenConvF16() argument
67 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF16()
|
| D | runtime_conv2d_mkl.cc | 54 int64 input_cols, int64 input_channels, int64 kernel_rows, in MKLConvImpl() argument
69 ToInt(input_rows), ToInt(input_cols)}; in MKLConvImpl()
156 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32() argument
167 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
173 MKLConvImpl(nullptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
|
| D | runtime_conv2d.h | 27 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
42 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
|
| D | runtime_single_threaded_conv2d.h | 28 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
41 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
|
| D | runtime_conv2d_impl.h | 34 Eigen::Index input_rows, Eigen::Index input_cols, in EigenConvImpl() argument
46 input(lhs, input_batch, input_rows, input_cols, input_channels); in EigenConvImpl()
|
| D | runtime_conv2d_mkl.h | 27 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 143 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
150 mkldnn_sizes[MklDnnDims::Dim_W] = input_cols; in GetInputSizeInMklOrder()
167 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
175 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_cols; in GetInputSizeInMklOrder()
330 int input_planes, input_rows, input_cols; variable
333 input_cols = GetTensorDim(input_shape, data_format_, 'W');
337 input_cols = GetTensorDim(input_shape, data_format_, '2');
417 input_cols, filter_cols, dilation_cols, stride_cols,
430 input_cols, filter_cols, dilation_cols, stride_cols,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_test.py | 857 input_cols=4,
879 input_cols=4,
1853 input_cols, argument
1867 input_shape = [batch, input_rows, input_cols, in_depth]
1872 output_cols = (input_cols - filter_cols + stride_cols) // stride_cols
1875 output_cols = (input_cols + stride_cols - 1) // stride_cols
1880 output_cols = (input_cols + padding[2][0] + padding[2][1] - filter_cols +
1947 input_cols=4,
1965 input_cols=5,
1983 input_cols=5,
[all …]
|
| D | conv_ops_3d_test.py | 467 input_planes, input_rows, input_cols = input_shape
470 input_shape = [batch, input_planes, input_rows, input_cols, in_depth]
486 math.ceil((input_cols - filter_cols + 1.0) / strides[3]))
490 output_cols = int(math.ceil(float(input_cols) / strides[3]))
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_image_patch.cpp | 187 int input_cols = 3; in test_patch_padding_valid() local
191 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches); in test_patch_padding_valid()
224 for (int j = 0; (j+stride+ksize-1) < input_cols; j += stride) { // input cols in test_patch_padding_valid()
234 … if (row_offset >= 0 && col_offset >= 0 && row_offset < input_rows && col_offset < input_cols) { in test_patch_padding_valid()
263 int input_cols = 5; in test_patch_padding_valid_same_value() local
268 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches); in test_patch_padding_valid_same_value()
297 for (int j = 0; (j+stride+ksize-1) <= input_cols; j += stride) { // input cols in test_patch_padding_valid_same_value()
307 … if (row_offset >= 0 && col_offset >= 0 && row_offset < input_rows && col_offset < input_cols) { in test_patch_padding_valid_same_value()
336 int input_cols = 2; in test_patch_padding_same() local
341 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches); in test_patch_padding_same()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/neon/ |
| D | neon_depthwise_conv_op.cc | 80 const int32 input_cols = input.dim_size(2); in Compute() local
95 GetWindowedOutputSize(input_cols, filter_cols, stride, in Compute()
110 << " Input: [" << batch << ", " << input_rows << ", " << input_cols in Compute()
|
| /external/libjpeg-turbo/ |
| D | jcsample.c | 94 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge() argument
101 int numcols = (int)(output_cols - input_cols); in expand_right_edge()
105 ptr = image_data[row] + input_cols; in expand_right_edge()
|