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

12

/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc33 const int input_rows = 3; local
38 const int output_rows = input_rows - patch_rows + 1;
41 Tensor<float, 3> input_backward(input_depth, input_rows, input_cols);
50 input_rows, input_cols, 1);
53 EXPECT_EQ(input_backward.dimension(1), input_rows);
57 for (int i = 0; i < input_rows; ++i) {
82 const int input_rows = 3; local
87 const int output_rows = input_rows - patch_rows + 1;
90 Tensor<float, 3, RowMajor> input_backward(input_cols, input_rows,
102 input_rows, input_col
135 const int input_rows = 3; local
198 const int input_rows = 3; local
260 const int input_rows = 7; local
308 const int input_rows = 7; local
360 const int input_rows = 3; local
431 const int input_rows = 3; local
502 const int input_rows = 7; local
557 const int input_rows = 7; local
614 const int input_rows = 3; local
681 const int input_rows = 3; 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
803 int input_rows = 3; 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
880 int input_rows = 3; local
898 const int input_rows = 3; local
948 const int input_rows = 3; local
1004 const int input_rows = 9; local
1051 const int input_rows = 9; local
1101 const int input_rows = 9; local
1149 const int input_rows = 9; local
1196 const int input_rows = 9; local
1241 const int input_rows = 3; local
1304 const int input_rows = 3; local
1368 const int input_rows = 7; local
1423 const int input_rows = 7; local
1479 const int input_rows = 7; local
1543 const int input_rows = 7; local
1610 const int input_rows = 7; local
1676 const int input_rows = 7; local
1736 const int input_rows = 7; local
1798 const int input_rows = 7; local
1877 const int input_rows = 7; local
1956 const int input_rows = 13; local
2033 const int input_rows = 13; local
[all...]
H A Ddilation_ops_gpu.cu.cc40 const T* filter_ptr, int batch, int input_rows,
59 if (h_in >= 0 && h_in < input_rows) {
65 input_cols * (h_in + input_rows * b))] +
81 const T* out_backprop_ptr, int batch, int input_rows, int input_cols,
103 if (h_in >= 0 && h_in < input_rows) {
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,
152 if (h_in >= 0 && h_in < input_rows) {
158 input_cols * (h_in + input_rows *
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
186 const int input_rows = input.dimension(1); local
217 const int input_rows = input.dimension(1); local
257 const int input_rows = input.dimension(1); local
[all...]
H A Deigen_pooling_test.cc30 const int input_rows = 5; local
38 Tensor<float, 4> input(depth, input_rows, input_cols, num_batches);
78 const int input_rows = 5; local
86 Tensor<float, 4, RowMajor> input(num_batches, input_cols, input_rows, depth);
128 const int input_rows = 5; local
138 Tensor<float, 5> input(channels, input_planes, input_rows, input_cols,
188 const int input_rows = 5; local
198 Tensor<float, 5, RowMajor> input(num_batches, input_cols, input_rows,
248 const int input_rows = 5; local
258 Tensor<float, 5> input(channels, input_planes, input_rows, input_col
310 const int input_rows = 5; local
372 const int input_rows = 5; local
450 const int input_rows = 5; local
527 const int input_rows = 5; local
575 const int input_rows = 5; local
625 const int input_rows = 5; local
685 const int input_rows = 5; local
[all...]
H A Deigen_spatial_convolutions_test.cc29 const int input_rows = 4; local
34 const int output_rows = input_rows;
37 Tensor<float, 3> input(input_depth, input_rows, input_cols);
74 const int input_rows = 4; local
79 const int output_rows = input_rows;
82 Tensor<float, 3, RowMajor> input(input_cols, input_rows, input_depth);
193 const int input_rows = 5; local
199 const int output_rows = input_rows - patch_rows + 1;
202 Tensor<float, 4> input(input_depth, input_rows, input_cols, num_batches);
245 const int input_rows local
303 const int input_rows = 5; local
359 const int input_rows = 5; local
407 const int input_rows = 3; local
455 const int input_rows = 5; local
506 const int input_rows = 7; local
557 const int input_rows = 7; local
611 const int input_rows = 7; local
[all...]
H A Ddilation_ops.cc70 // [ batch, input_rows, input_cols, depth ]
75 const int input_rows = input.dim_size(1); local
107 context, GetWindowedOutputSize(input_rows, filter_rows_eff, *stride_rows,
169 const int input_rows = input.dimension(1); local
190 if (h_in >= 0 && h_in < input_rows) {
245 // [ batch, input_rows, input_cols, depth ]
278 const int input_rows = input.dimension(1); local
307 if (h_in >= 0 && h_in < input_rows) {
365 // [ batch, input_rows, input_cols, depth ]
398 const int input_rows local
[all...]
H A Dconv_ops.cc136 const Tensor& filter, int batch, int input_rows,
152 const Tensor& filter, int batch, int input_rows,
167 args.in_rows = input_rows;
193 const Tensor& filter, int batch, int input_rows,
207 const Tensor& filter, int batch, int input_rows,
219 desc.H = input_rows;
344 const int input_rows = static_cast<int>(input_rows_raw); variable
375 input_rows, filter_rows, dilation_rows,
391 << ", input_rows = " << input_rows
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.cc495 const int input_rows = static_cast<int>(input_rows_raw); variable
522 GetWindowedOutputSize(input_rows, filter_rows, stride_rows,
538 << ", input_rows = " << input_rows
549 conv_functor(context, input.flat<T>().data(), batch, input_rows, input_cols,
H A Ddepthwise_conv_op.cc327 const int32 input_rows = static_cast<int32>(input_rows_raw); variable
343 GetWindowedOutputSize(input_rows, filter_rows, stride_,
361 << " Input: [" << batch << ", " << input_rows << ", " << input_cols
386 args.in_rows = input_rows;
H A Dmkl_conv_ops.h103 int input_rows = static_cast<int>(input_rows_raw); local
121 mkldnn_sizes[MklDnnDims::Dim_H] = input_rows;
224 int input_rows = GetTensorDim(input_shape, data_format_, 'H'); local
245 input_rows, filter_rows, stride_rows, padding_,
H A Dquantized_conv_ops.cc524 const int64 input_rows = input.dim_size(1); variable
542 GetWindowedOutputSize(input_rows, filter_rows, stride,
561 conv_functor(context, input.flat<T1>().data(), batch, input_rows,
H A Ddepthwise_conv_grad_op.cc72 const int32 input_rows = static_cast<int32>(input_rows_raw); \
114 GetWindowedOutputSize(input_rows, filter_rows, stride, \
131 args.in_rows = input_rows; \
144 << input_rows << ", " << input_cols << ", " << in_depth \
H A Dmkl_conv_ops.cc144 const int input_rows = static_cast<int>(input_rows_raw); variable
176 GetWindowedOutputSize(input_rows, filter_rows, stride_rows,
214 mkl_context.in_sizes[MklDims::H] = static_cast<size_t>(input_rows);
H A Dconv_grad_filter_ops.cc116 int input_rows, int input_cols, int row_stride,
128 typename TTypes<float, 4>::ConstTensor output, int input_rows,
143 desc.H = input_rows;
112 operator ()(OpKernelContext* context, const Device& d, typename TTypes<T, 4>::ConstTensor input_backward, typename TTypes<T, 4>::Tensor kernel, typename TTypes<T, 4>::ConstTensor output_backward, int input_rows, int input_cols, int row_stride, int col_stride, int pad_h, int pad_w, TensorFormat data_format) const argument
125 operator ()(OpKernelContext* context, const CPUDevice& d, typename TTypes<float, 4>::ConstTensor input, typename TTypes<float, 4>::Tensor filter, typename TTypes<float, 4>::ConstTensor output, int input_rows, int input_cols, int row_stride, int col_stride, int pad_h, int pad_w, TensorFormat data_format) const argument
/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,
35 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, local
45 int64 input_batch, int64 input_rows, int64 input_cols, int64 input_channels,
52 Eigen::DefaultDevice(), out, lhs, rhs, input_batch, input_rows, 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,
40 tensorflow::int64 input_batch, tensorflow::int64 input_rows,
H A Druntime_single_threaded_conv2d.h26 tensorflow::int64 input_batch, tensorflow::int64 input_rows,
40 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
H A Druntime_conv2d_impl.h29 ScalarType* rhs, int64 input_batch, int64 input_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/libjpeg-turbo/
H A Djcprepct.c104 * Expand an image vertically from height input_rows to height output_rows,
110 int input_rows, int output_rows)
114 for (row = input_rows; row < output_rows; row++) {
115 jcopy_sample_rows(image_data, input_rows-1, image_data, row,
109 expand_bottom_edge(JSAMPARRAY image_data, JDIMENSION num_cols, int input_rows, int output_rows) argument
/external/tensorflow/tensorflow/core/kernels/neon/
H A Dneon_depthwise_conv_op.cc79 const int32 input_rows = input.dim_size(1); variable
92 GetWindowedOutputSize(input_rows, filter_rows, stride,
110 << " Input: [" << batch << ", " << input_rows << ", " << input_cols
/external/eigen/unsupported/test/
H A Dcxx11_tensor_image_patch.cpp186 int input_rows = 3; local
191 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
223 for (int i = 0; (i+stride+ksize-1) < input_rows; i += stride) { // input rows
225 int patchId = i+input_rows*j;
234 if (row_offset >= 0 && col_offset >= 0 && row_offset < input_rows && col_offset < input_cols) {
262 int input_rows = 5; local
268 Tensor<float, 4> tensor(input_depth, input_rows, input_cols, input_batches);
296 for (int i = 0; (i+stride+ksize-1) <= input_rows; i += stride) { // input rows
298 int patchId = i+input_rows*j;
307 if (row_offset >= 0 && col_offset >= 0 && row_offset < input_rows
335 int input_rows = 4; local
[all...]
/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]
1123 output_rows = (input_rows - filter_rows + stride_rows) // stride_rows
1126 output_rows = (input_rows + stride_rows - 1) // stride_rows
1188 input_rows=5,
1205 input_rows=6,
1222 input_rows=4,
1239 input_rows
[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]
295 math.ceil((input_rows - filter_rows + 1.0) / strides[2]))
300 output_rows = int(math.ceil(float(input_rows) / strides[2]))
/external/tensorflow/tensorflow/contrib/factorization/python/ops/
H A Dwals.py181 input_rows = features[WALSMatrixFactorization.INPUT_ROWS]
281 input_rows,
324 sp_input=input_rows,
353 sp_input=input_rows, transpose_input=False)
442 INPUT_ROWS = "input_rows"
H A Dfactorization_ops.py187 input_rows,
203 input_rows: total number of rows for input matrix.
242 self._input_rows = input_rows

Completed in 432 milliseconds

12