/external/libjpeg-turbo/simd/ |
H A D | jcsample.h | 12 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 D | eigen_backward_spatial_convolutions_test.cc | 34 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 D | dilation_ops_gpu.cu.cc | 41 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 D | eigen_pooling_test.cc | 31 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 D | eigen_spatial_convolutions_test.cc | 30 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 D | dilation_ops.cc | 70 // [ 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 D | conv_ops.cc | 137 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 D | conv_ops_using_gemm.cc | 505 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 D | depthwise_conv_op.cc | 335 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 D | mkl_conv_ops.h | 108 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 D | quantized_conv_ops.cc | 529 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 D | depthwise_conv_grad_op.cc | 78 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 D | mkl_conv_ops.cc | 156 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 D | runtime_single_threaded_conv2d.cc | 27 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 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, 41 tensorflow::int64 input_cols, tensorflow::int64 input_channels,
|
H A D | runtime_single_threaded_conv2d.h | 27 tensorflow::int64 input_cols, tensorflow::int64 input_channels, 40 tensorflow::int64 input_rows, tensorflow::int64 input_cols,
|
H A D | runtime_conv2d_impl.h | 30 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 D | neon_depthwise_conv_op.cc | 80 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 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] 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 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] 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 D | cxx11_tensor_image_patch.cpp | 187 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 D | wals.py | 182 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 D | factorization_ops.py | 188 input_cols, 204 input_cols: total number of cols for input matrix. 243 self._input_cols = input_cols
|
/external/libjpeg-turbo/ |
H A D | jcsample.c | 89 * 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
|