/external/tensorflow/tensorflow/core/kernels/ |
H A D | eigen_backward_spatial_convolutions_test.cc | 33 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 D | dilation_ops_gpu.cu.cc | 40 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 D | eigen_pooling_test.cc | 30 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 D | eigen_spatial_convolutions_test.cc | 29 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 D | dilation_ops.cc | 70 // [ 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 D | conv_ops.cc | 136 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 D | conv_ops_using_gemm.cc | 495 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 D | depthwise_conv_op.cc | 327 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 D | mkl_conv_ops.h | 103 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 D | quantized_conv_ops.cc | 524 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 D | depthwise_conv_grad_op.cc | 72 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 D | mkl_conv_ops.cc | 144 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 D | conv_grad_filter_ops.cc | 116 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 D | runtime_single_threaded_conv2d.cc | 27 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 D | runtime_conv2d.cc | 29 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 D | runtime_conv2d.h | 26 tensorflow::int64 input_rows, tensorflow::int64 input_cols, 40 tensorflow::int64 input_batch, tensorflow::int64 input_rows,
|
H A D | runtime_single_threaded_conv2d.h | 26 tensorflow::int64 input_batch, tensorflow::int64 input_rows, 40 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
|
H A D | runtime_conv2d_impl.h | 29 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 D | jcprepct.c | 104 * 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 D | neon_depthwise_conv_op.cc | 79 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 D | cxx11_tensor_image_patch.cpp | 186 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 D | conv_ops_test.py | 174 [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 D | conv_ops_3d_test.py | 278 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 D | wals.py | 181 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 D | factorization_ops.py | 187 input_rows, 203 input_rows: total number of rows for input matrix. 242 self._input_rows = input_rows
|