Searched refs:input_cols (Results 1 - 25 of 30) sorted by relevance

12

/external/libjpeg-turbo/simd/
H A Djcsample.h12 JDIMENSION input_cols, JDIMENSION output_cols)
18 int numcols = (int) (output_cols - input_cols);
22 ptr = image_data[row] + input_cols;
11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, JDIMENSION output_cols) argument
/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc34 const int input_cols = 4; local
39 const int output_cols = input_cols - patch_cols + 1;
41 Tensor<float, 3> input_backward(input_depth, input_rows, input_cols);
50 input_rows, input_cols, 1);
54 EXPECT_EQ(input_backward.dimension(2), input_cols);
58 for (int j = 0; j < input_cols; ++j) {
83 const int input_cols = 4; local
88 const int output_cols = input_cols - patch_cols + 1;
90 Tensor<float, 3, RowMajor> input_backward(input_cols, input_rows,
102 input_rows, input_cols,
136 const int input_cols = 4; local
199 const int input_cols = 4; local
261 const int input_cols = 9; local
309 const int input_cols = 9; local
361 const int input_cols = 4; local
432 const int input_cols = 4; local
503 const int input_cols = 9; local
558 const int input_cols = 9; local
615 const int input_cols = 4; local
682 const int input_cols = 4; local
743 test_batched_strided_spatial_convolution_backward_input_valid( const int num_batches, const int input_depth, const int input_rows, const int input_cols, const int output_depth) argument
804 int input_cols = 5; local
819 test_batched_strided_spatial_convolution_backward_input_valid_row_major( const int num_batches, const int input_depth, const int input_rows, const int input_cols, const int output_depth) argument
881 int input_cols = 5; local
899 const int input_cols = 4; local
949 const int input_cols = 4; local
1005 const int input_cols = 13; local
1052 const int input_cols = 13; local
1102 const int input_cols = 13; local
1150 const int input_cols = 13; local
1197 const int input_cols = 13; local
1242 const int input_cols = 4; local
1305 const int input_cols = 4; local
1369 const int input_cols = 9; local
1424 const int input_cols = 9; local
1480 const int input_cols = 9; local
1544 const int input_cols = 9; local
1611 const int input_cols = 9; local
1677 const int input_cols = 9; local
1737 const int input_cols = 9; local
1799 const int input_cols = 9; local
1878 const int input_cols = 9; local
1957 const int input_cols = 15; local
2034 const int input_cols = 15; local
[all...]
H A Ddilation_ops_gpu.cu.cc41 int input_cols, int depth, int filter_rows,
62 if (w_in >= 0 && w_in < input_cols) {
65 input_cols * (h_in + input_rows * b))] +
81 const T* out_backprop_ptr, int batch, int input_rows, int input_cols,
106 if (w_in >= 0 && w_in < input_cols) {
109 input_cols * (h_in + input_rows * b))] +
122 depth * (w_in_max + input_cols * (h_in_max + input_rows * b)),
130 const T* out_backprop_ptr, int batch, int input_rows, int input_cols,
155 if (w_in >= 0 && w_in < input_cols) {
158 input_cols * (h_i
39 DilationKernel(const int32 nthreads, const T* input_ptr, const T* filter_ptr, int batch, int input_rows, int input_cols, int depth, int filter_rows, int filter_cols, int output_rows, int output_cols, int stride_rows, int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, T* output_ptr) argument
79 DilationBackpropInputKernel( const int32 nthreads, const T* input_ptr, const T* filter_ptr, const T* out_backprop_ptr, int batch, int input_rows, int input_cols, int depth, int filter_rows, int filter_cols, int output_rows, int output_cols, int stride_rows, int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, T* in_backprop_ptr) argument
128 DilationBackpropFilterKernel( const int32 nthreads, const T* input_ptr, const T* filter_ptr, const T* out_backprop_ptr, int batch, int input_rows, int input_cols, int depth, int filter_rows, int filter_cols, int output_rows, int output_cols, int stride_rows, int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, T* filter_backprop_ptr) argument
187 const int input_cols = input.dimension(2); local
218 const int input_cols = input.dimension(2); local
258 const int input_cols = input.dimension(2); local
[all...]
H A Deigen_pooling_test.cc31 const int input_cols = 5; local
38 Tensor<float, 4> input(depth, input_rows, input_cols, num_batches);
79 const int input_cols = 5; local
86 Tensor<float, 4, RowMajor> input(num_batches, input_cols, input_rows, depth);
129 const int input_cols = 5; local
138 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols,
189 const int input_cols = 5; local
198 Tensor<float, 5, RowMajor> input(num_batches, input_cols, input_rows,
249 const int input_cols = 5; local
258 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols,
311 const int input_cols = 5; local
373 const int input_cols = 5; local
451 const int input_cols = 5; local
528 const int input_cols = 5; local
576 const int input_cols = 5; local
626 const int input_cols = 5; local
686 const int input_cols = 5; local
[all...]
H A Deigen_spatial_convolutions_test.cc30 const int input_cols = 5; local
35 const int output_cols = input_cols;
37 Tensor<float, 3> input(input_depth, input_rows, input_cols);
75 const int input_cols = 5; local
80 const int output_cols = input_cols;
82 Tensor<float, 3, RowMajor> input(input_cols, input_rows, input_depth);
194 const int input_cols = 5; local
200 const int output_cols = input_cols - patch_cols + 1;
202 Tensor<float, 4> input(input_depth, input_rows, input_cols, num_batches);
246 const int input_cols local
304 const int input_cols = 5; local
360 const int input_cols = 5; local
408 const int input_cols = 3; local
456 const int input_cols = 5; local
507 const int input_cols = 7; local
558 const int input_cols = 7; local
612 const int input_cols = 7; local
[all...]
H A Ddilation_ops.cc70 // [ batch, input_rows, input_cols, depth ]
76 const int input_cols = input.dim_size(2); local
110 context, GetWindowedOutputSize(input_cols, filter_cols_eff, *stride_cols,
170 const int input_cols = input.dimension(2); local
193 if (w_in >= 0 && w_in < input_cols) {
245 // [ batch, input_rows, input_cols, depth ]
279 const int input_cols = input.dimension(2); local
310 if (w_in >= 0 && w_in < input_cols) {
365 // [ batch, input_rows, input_cols, depth ]
399 const int input_cols local
[all...]
H A Dconv_ops.cc137 int input_cols, int in_depth, int filter_rows,
153 int input_cols, int in_depth, int filter_rows,
168 args.in_cols = input_cols;
194 int input_cols, int in_depth, int filter_rows,
208 int input_cols, int in_depth, int filter_rows,
220 desc.W = input_cols;
354 const int input_cols = static_cast<int>(input_cols_raw); variable
378 input_cols, filter_cols, dilation_cols,
389 << ", input_cols = " << input_cols
135 Run(OpKernelContext* ctx, const Tensor& input, const Tensor& filter, int batch, int input_rows, int input_cols, int in_depth, int filter_rows, int filter_cols, int pad_rows, int pad_cols, int out_rows, int , int , int , int , int , int , Tensor* , TensorFormat ) argument
151 Run(OpKernelContext* ctx, const Tensor& input, const Tensor& filter, int batch, int input_rows, int input_cols, int in_depth, int filter_rows, int filter_cols, int pad_rows, int pad_cols, int out_rows, int out_cols, int out_depth, int dilation_rows, int dilation_cols, int stride_rows, int stride_cols, Tensor* output, TensorFormat data_format) argument
192 Run(OpKernelContext* ctx, const Tensor& input, const Tensor& filter, int batch, int input_rows, int input_cols, int in_depth, int filter_rows, int filter_cols, int pad_rows, int pad_cols, int out_rows, int out_cols, int out_depth, int stride_rows, int stride_cols, int dilation_rows, int dilation_cols, Tensor* output, TensorFormat data_format) argument
206 Run(OpKernelContext* ctx, const Tensor& input, const Tensor& filter, int batch, int input_rows, int input_cols, int in_depth, int filter_rows, int filter_cols, int pad_rows, int pad_cols, int out_rows, int out_cols, int out_depth, int dilation_rows, int dilation_cols, int stride_rows, int stride_cols, Tensor* output, TensorFormat data_format) argument
[all...]
H A Dconv_ops_using_gemm.cc505 const int input_cols = static_cast<int>(input_cols_raw); variable
525 GetWindowedOutputSize(input_cols, filter_cols, stride_cols,
536 << ", input_cols = " << input_cols
549 conv_functor(context, input.flat<T>().data(), batch, input_rows, input_cols,
H A Ddepthwise_conv_op.cc335 const int32 input_cols = static_cast<int32>(input_cols_raw); variable
346 GetWindowedOutputSize(input_cols, filter_cols, stride_,
361 << " Input: [" << batch << ", " << input_rows << ", " << input_cols
387 args.in_cols = input_cols;
H A Dmkl_conv_ops.h108 int input_cols = static_cast<int>(input_cols_raw); local
122 mkldnn_sizes[MklDnnDims::Dim_W] = input_cols;
225 int input_cols = GetTensorDim(input_shape, data_format_, 'W'); local
248 input_cols, filter_cols, stride_cols, padding_,
H A Dquantized_conv_ops.cc529 const int64 input_cols = input.dim_size(2); variable
545 GetWindowedOutputSize(input_cols, filter_cols, stride,
562 input_cols, in_depth, offset_input, filter.flat<T2>().data(),
H A Ddepthwise_conv_grad_op.cc78 const int32 input_cols = static_cast<int32>(input_cols_raw); \
117 GetWindowedOutputSize(input_cols, filter_cols, stride, \
132 args.in_cols = input_cols; \
144 << input_rows << ", " << input_cols << ", " << in_depth \
H A Dmkl_conv_ops.cc156 const int input_cols = static_cast<int>(input_cols_raw); variable
179 GetWindowedOutputSize(input_cols, filter_cols, stride_cols,
213 mkl_context.in_sizes[MklDims::W] = static_cast<size_t>(input_cols);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_single_threaded_conv2d.cc27 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols,
36 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, local
45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels,
53 input_cols, input_channels, kernel_rows, kernel_cols, kernel_channels, local
25 __xla_cpu_runtime_EigenSingleThreadedConvF16( const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, int64 output_rows, int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, int64 padding_left, int64 padding_right, int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) argument
43 __xla_cpu_runtime_EigenSingleThreadedConvF32( const void* run_options_ptr, float* out, float* lhs, float* rhs, int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, int64 output_rows, int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, int64 padding_left, int64 padding_right, int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) argument
H A Druntime_conv2d.cc29 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels,
39 input_rows, input_cols, input_channels, kernel_rows, kernel_cols,
47 Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols,
58 input_rows, input_cols, input_channels, kernel_rows, kernel_cols,
27 __xla_cpu_runtime_EigenConvF32( const void* run_options_ptr, float* out, float* lhs, float* rhs, int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, int64 output_rows, int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, int64 padding_left, int64 padding_right, int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) argument
45 __xla_cpu_runtime_EigenConvF16( const void* run_options_ptr, Eigen::half* out, Eigen::half* lhs, Eigen::half* rhs, int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, int64 output_rows, int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, int64 padding_left, int64 padding_right, int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) argument
H A Druntime_conv2d.h26 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
41 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
H A Druntime_single_threaded_conv2d.h27 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
40 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
H A Druntime_conv2d_impl.h30 int64 input_cols, int64 input_channels, int64 kernel_rows,
40 input(lhs, input_batch, input_rows, input_cols, input_channels);
28 EigenConvImpl(const EigenDevice& device, ScalarType* out, ScalarType* lhs, ScalarType* rhs, int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels, int64 kernel_rows, int64 kernel_cols, int64 kernel_channels, int64 kernel_filters, int64 output_rows, int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, int64 padding_left, int64 padding_right, int64 lhs_row_dilation, int64 lhs_col_dilation, int64 rhs_row_dilation, int64 rhs_col_dilation) argument
/external/tensorflow/tensorflow/core/kernels/neon/
H A Dneon_depthwise_conv_op.cc80 const int32 input_cols = input.dim_size(2); variable
95 GetWindowedOutputSize(input_cols, filter_cols, stride,
110 << " Input: [" << batch << ", " << input_rows << ", " << input_cols
/external/tensorflow/tensorflow/python/kernel_tests/
H A Dconv_ops_test.py174 [batch, input_rows, input_cols, input_depth].
224 [batch, input_rows, input_cols, input_depth].
1115 def ConstructAndTestGradient(self, batch, input_rows, input_cols, filter_rows,
1119 input_shape = [batch, input_rows, input_cols, in_depth]
1124 output_cols = (input_cols - filter_cols + stride_cols) // stride_cols
1127 output_cols = (input_cols + stride_cols - 1) // stride_cols
1189 input_cols=4,
1206 input_cols=5,
1223 input_cols=5,
1240 input_cols
[all...]
H A Dconv_ops_3d_test.py278 input_planes, input_rows, input_cols = input_shape
281 input_shape = [batch, input_planes, input_rows, input_cols, in_depth]
297 math.ceil((input_cols - filter_cols + 1.0) / strides[3]))
301 output_cols = int(math.ceil(float(input_cols) / strides[3]))
/external/eigen/unsupported/test/
H A Dcxx11_tensor_image_patch.cpp187 int input_cols = 3; local
191 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
224 for (int j = 0; (j+stride+ksize-1) < input_cols; j += stride) { // input cols
234 if (row_offset >= 0 && col_offset >= 0 && row_offset < input_rows && col_offset < input_cols) {
263 int input_cols = 5; local
268 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
297 for (int j = 0; (j+stride+ksize-1) <= input_cols; j += stride) { // input cols
307 if (row_offset >= 0 && col_offset >= 0 && row_offset < input_rows && col_offset < input_cols) {
336 int input_cols = 2; local
341 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batche
[all...]
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
H A Dwals.py182 input_cols = features[WALSMatrixFactorization.INPUT_COLS]
286 input_cols,
330 sp_input=input_cols,
357 sp_input=input_cols, transpose_input=True)
443 INPUT_COLS = "input_cols"
H A Dfactorization_ops.py188 input_cols,
204 input_cols: total number of cols for input matrix.
243 self._input_cols = input_cols
/external/libjpeg-turbo/
H A Djcsample.c89 * Expand a component horizontally from width input_cols to width output_cols,
95 JDIMENSION input_cols, JDIMENSION output_cols)
101 int numcols = (int) (output_cols - input_cols);
105 ptr = image_data[row] + input_cols;
94 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, JDIMENSION output_cols) argument

Completed in 493 milliseconds

12