Home
last modified time | relevance | path

Searched refs:input_cols (Results 1 – 25 of 37) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Deigen_backward_spatial_convolutions_test.cc33 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 …]
Ddilation_ops_gpu.cu.cc41 int input_cols, int depth, int filter_rows, in DilationKernel() argument
62 if (w_in >= 0 && w_in < input_cols) { in DilationKernel()
65 input_cols * (h_in + input_rows * b))] + in DilationKernel()
81 const T* out_backprop_ptr, int batch, int input_rows, int input_cols, in DilationBackpropInputKernel() argument
106 if (w_in >= 0 && w_in < input_cols) { in DilationBackpropInputKernel()
109 input_cols * (h_in + input_rows * b))] + in DilationBackpropInputKernel()
122 depth * (w_in_max + input_cols * (h_in_max + input_rows * b)), in DilationBackpropInputKernel()
130 const T* out_backprop_ptr, int batch, int input_rows, int input_cols, 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 …]
Deigen_pooling_test.cc30 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 …]
Deigen_spatial_convolutions_test.cc32 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 …]
Dmkl_conv_ops.h137 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
144 mkldnn_sizes[MklDnnDims::Dim_W] = input_cols; in GetInputSizeInMklOrder()
161 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
169 mkldnn_sizes[MklDnnDims3D::Dim3d_W] = input_cols; in GetInputSizeInMklOrder()
324 int input_planes, input_rows, input_cols; variable
327 input_cols = GetTensorDim(input_shape, data_format_, 'W');
331 input_cols = GetTensorDim(input_shape, data_format_, '2');
411 input_cols, filter_cols, dilation_cols, stride_cols,
421 input_cols, filter_cols, stride_cols,
Ddilation_ops.cc76 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 ()()
Dconv_ops.cc156 int input_cols, int in_depth, int filter_rows, in Run() argument
172 int input_cols, int in_depth, int filter_rows, in Run() argument
187 args.in_cols = input_cols; in Run()
213 int input_cols, int in_depth, int filter_rows, in Run() argument
227 int input_cols, int in_depth, int filter_rows, in Run() argument
239 desc.W = input_cols; in Run()
382 const int input_cols = static_cast<int>(input_cols_raw); in ComputeConv2DDimension() local
414 input_cols, filter_cols, dilation_cols, stride_cols, params.padding, in ComputeConv2DDimension()
419 dimensions->input_cols = input_cols; in ComputeConv2DDimension()
476 << ", input_cols = " << dimensions.input_cols in Compute()
[all …]
Deigen_benchmark.h89 Eigen::Index input_cols = input_dims[2]; in SpatialConvolutionBackwardInput() local
108 filter, output_backward, input_rows, input_cols); in SpatialConvolutionBackwardInput()
226 Eigen::Index input_cols = input_dims[2]; in CuboidConvolutionBackwardInput() local
246 filter, output_backward, input_planes, input_rows, input_cols); in CuboidConvolutionBackwardInput()
Dconv_ops_using_gemm.cc505 const int input_cols = static_cast<int>(input_cols_raw); in Compute() local
525 GetWindowedOutputSize(input_cols, filter_cols, stride_cols, in Compute()
536 << ", input_cols = " << input_cols in Compute()
549 conv_functor(context, input.flat<T>().data(), batch, input_rows, input_cols, in Compute()
Ddepthwise_conv_op.cc342 const int32 input_cols = static_cast<int32>(input_cols_raw); in Compute() local
353 GetWindowedOutputSize(input_cols, filter_cols, stride_, in Compute()
378 << " Input: [" << batch << ", " << input_rows << ", " << input_cols in Compute()
415 args.in_cols = input_cols; in Compute()
/external/libjpeg-turbo/simd/loongson/
Djcsample.h11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge() argument
18 int numcols = (int)(output_cols - input_cols); in expand_right_edge()
22 ptr = image_data[row] + 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() argument
18 int numcols = (int)(output_cols - input_cols); in expand_right_edge()
22 ptr = image_data[row] + 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() argument
36 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
53 input_cols, input_channels, kernel_rows, kernel_cols, kernel_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() argument
39 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF32()
47 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, in __xla_cpu_runtime_EigenConvF16() argument
58 input_rows, input_cols, input_channels, kernel_rows, kernel_cols, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc54 int64 input_cols, int64 input_channels, int64 kernel_rows, in MKLConvImpl() argument
69 ToInt(input_rows), ToInt(input_cols)}; in MKLConvImpl()
154 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, in __xla_cpu_runtime_MKLConvF32() argument
165 run_options_ptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
171 MKLConvImpl(nullptr, out, lhs, rhs, input_batch, input_rows, input_cols, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h28 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
41 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
Druntime_conv2d.h27 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
42 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
Druntime_conv2d_impl.h30 int64 input_cols, int64 input_channels, int64 kernel_rows, in EigenConvImpl() argument
40 input(lhs, input_batch, input_rows, input_cols, input_channels); in EigenConvImpl()
Druntime_conv2d_mkl.h27 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1638 def ConstructAndTestGradient(self, batch, input_rows, input_cols, filter_rows, argument
1642 input_shape = [batch, input_rows, input_cols, in_depth]
1647 output_cols = (input_cols - filter_cols + stride_cols) // stride_cols
1650 output_cols = (input_cols + stride_cols - 1) // stride_cols
1655 output_cols = (input_cols + padding[2][0] + padding[2][1] - filter_cols +
1722 input_cols=4,
1740 input_cols=5,
1758 input_cols=5,
1776 input_cols=5,
1794 input_cols=6,
[all …]
Dconv_ops_3d_test.py371 input_planes, input_rows, input_cols = input_shape
374 input_shape = [batch, input_planes, input_rows, input_cols, in_depth]
390 math.ceil((input_cols - filter_cols + 1.0) / strides[3]))
394 output_cols = int(math.ceil(float(input_cols) / strides[3]))
/external/eigen/unsupported/test/
Dcxx11_tensor_image_patch.cpp187 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/
Dneon_depthwise_conv_op.cc80 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/tensorflow/tensorflow/contrib/factorization/python/ops/
Dwals.py182 input_cols = features[WALSMatrixFactorization.INPUT_COLS]
286 input_cols,
330 sp_input=input_cols,
357 sp_input=input_cols, transpose_input=True)
/external/libjpeg-turbo/
Djcsample.c94 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()

12