Home
last modified time | relevance | path

Searched refs:col_stride (Results 1 – 25 of 43) sorted by relevance

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py951 col_stride, argument
994 strides = [1, 1, row_stride, col_stride]
998 strides = [1, row_stride, col_stride, 1]
1027 col_stride, argument
1070 strides = [1, 1, row_stride, col_stride]
1074 strides = [1, row_stride, col_stride, 1]
1106 col_stride=1,
1120 col_stride=1,
1134 col_stride=1,
1148 col_stride=2,
[all …]
/external/gemmlowp/internal/
Doutput_neon.h307 int col_stride = dst->cols_stride();
309 vst1q_s32(dst_ptr + i * col_stride + 0, src.buf.reg[2 * i + 0]);
310 vst1q_s32(dst_ptr + i * col_stride + 4, src.buf.reg[2 * i + 1]);
378 int col_stride = dst->cols_stride();
380 vst1q_s32(dst_ptr + i * col_stride, src.buf.reg[2 * i]);
381 vst1q_s32(dst_ptr + i * col_stride + 4, src.buf.reg[2 * i + 1]);
452 int col_stride = dst->cols_stride();
453 vst1q_lane_s32(dst_ptr + 0 * col_stride, src.buf.reg[0], 0);
454 vst1q_lane_s32(dst_ptr + 1 * col_stride, src.buf.reg[0], 1);
455 vst1q_lane_s32(dst_ptr + 2 * col_stride, src.buf.reg[0], 2);
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h57 Filter filter, int row_stride, int col_stride, in SpatialConvolutionFunc() argument
63 Eigen::SpatialConvolution(input, filter, col_stride, row_stride, padding, in SpatialConvolutionFunc()
73 int col_stride, int row_dilation, int col_dilation, in operator()
76 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator()
87 int row_stride, int col_stride, int row_dilation,
92 col_stride, row_stride, padding, col_dilation,
103 int row_stride, int col_stride, int row_dilation,
108 input_backward.dimension(1), col_stride, row_stride, col_dilation,
119 int row_stride, int col_stride, int row_dilation,
124 kernel_backward.dimension(0), col_stride, row_stride, col_dilation,
Dpooling_ops_common.h61 int col_stride; member
169 params.row_stride, params.col_stride, pt); in SpatialMaxPool()
205 const int32 col_stride = params.col_stride; in SpatialMaxPool() local
231 : (wpad - window_cols) / col_stride + 1; in SpatialMaxPool()
232 const int32 w_end = std::min(wpad / col_stride + 1, out_width); in SpatialMaxPool()
276 params.window_cols, params.row_stride, params.col_stride,
413 params.row_stride, params.col_stride, pt);
452 const int32 col_stride = params.col_stride;
478 : (wpad - window_cols) / col_stride + 1;
479 const int32 w_end = std::min(wpad / col_stride + 1, out_width);
[all …]
Dmaxpooling_op.h33 int window_cols, int row_stride, int col_stride, in operator()
38 col_stride, row_stride, padding); in operator()
46 int window_cols, int row_stride, int col_stride,
Deigen_backward_spatial_convolutions.h102 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
154 0, ((outputCols - 1) * col_stride + kernelColsEff - inputCols) / 2);
160 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride -
281 col_in_stride, row_stride, col_stride, padding_top,
288 col_in_stride, row_stride, col_stride,
352 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
370 !(col_stride > 1 && col_in_stride > 1));
412 0, (outputCols - 1) * col_stride + kernelColsEff - inputCols);
484 kernelRows, kernelCols, row_stride, col_stride,
491 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
Davgpooling_op.h31 int window_cols, int row_stride, int col_stride, in operator()
36 To32Bit(input).swap_layout(), window_cols, window_rows, col_stride, in operator()
41 input.swap_layout(), window_cols, window_rows, col_stride, row_stride, in operator()
Dconv_grad_ops.h179 int col_stride, const Padding& padding,
189 int col_stride, const Padding& padding,
199 int col_dilation, int row_stride, int col_stride,
210 int col_stride, const Padding& padding,
Davgpooling_op.cc171 params.row_stride, params.col_stride, pt); in Compute()
189 int window_cols, int row_stride, int col_stride, \
276 const int col_stride = stride_[2]; in Compute() local
291 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
300 col_stride, pad_rows, pad_cols](int64 start, int64 limit) { in Compute()
320 GetBroadcastSize(c, in_cols, window_cols, col_stride, in Compute()
516 const int col_stride = stride_[2]; in Compute() local
531 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
544 col_stride, // stride_w in Compute()
Dconv_ops.cc71 const Tensor& filter, int row_stride, int col_stride, in operator ()()
77 col_stride == 1) { in operator ()()
115 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
126 int col_dilation, int row_stride, int col_stride, in operator ()()
145 LaunchGeneric<CPUDevice, T>()(ctx, input, filter, row_stride, col_stride, in operator ()()
581 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
612 col_stride == 1 && data_format == FORMAT_NHWC && in operator ()()
688 col_stride, padding, &out_cols_check, in operator ()()
784 .set_horizontal_filter_stride(col_stride) in operator ()()
841 col_stride}}, // stride_cols in operator ()()
[all …]
Dconv_ops.h38 int col_dilation, int row_stride, int col_stride,
49 int col_dilation, int row_stride, int col_stride,
Dops_util_test.cc38 int col_stride; member
79 pad_struct.input.col_stride, pad_struct.input.padding, &new_width, in VerifyGet2dOutputSizeBoundaries()
94 pad_struct.input.col_stride, pad_struct.input.padding, &new_width, in VerifyGet2dOutputSizeValues()
113 pad_struct.input.col_stride, pad_struct.input.padding, &new_width, in VerifyGet2dOutputVerboseSizeValues()
Dpooling_ops_common.cc70 col_stride = GetTensorDim(stride, data_format, 'W'); in PoolParameters()
86 context, GetWindowedOutputSize(tensor_in_cols, window_cols, col_stride, in PoolParameters()
233 .set_horizontal_stride(params.col_stride) in Compute()
381 .set_horizontal_stride(params.col_stride) in Compute()
Deigen_backward_cuboid_convolutions.h93 const DenseIndex col_stride = 1) {
149 0, ((outputCols - 1) * col_stride + kernelColsEff - inputCols) / 2);
160 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride -
283 plane_stride, row_stride, col_stride, padding_top_z,
291 plane_stride, row_stride, col_stride,
Dconv_grad_filter_ops.cc106 int col_stride, const Padding& padding, in operator ()()
112 out_backprop.tensor<T, 4>(), row_stride, col_stride, in operator ()()
125 int col_stride, int pad_h, int pad_w, in operator ()()
137 int input_cols, int row_stride, int col_stride, int pad_h, in operator ()()
157 desc.v = col_stride; in operator ()()
546 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
559 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
587 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc30 int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
37 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
48 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
54 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc32 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_EigenConvF32() argument
41 col_stride, padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenConvF32()
50 int64 output_cols, int64 row_stride, int64 col_stride, int64 padding_top, in __xla_cpu_runtime_EigenConvF16() argument
60 col_stride, padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenConvF16()
Druntime_conv2d_mkl.cc57 int64 col_stride, int64 padding_top, int64 padding_bottom, in MKLConvImpl() argument
75 memory::dims conv1_strides = {ToInt(row_stride), ToInt(col_stride)}; in MKLConvImpl()
157 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_MKLConvF32() argument
167 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
174 col_stride, padding_top, padding_bottom, padding_left, in __xla_cpu_runtime_MKLConvF32()
Druntime_single_threaded_conv2d.h32 tensorflow::int64 row_stride, tensorflow::int64 col_stride,
46 tensorflow::int64 col_stride, tensorflow::int64 padding_top,
Druntime_conv2d.h32 tensorflow::int64 col_stride, tensorflow::int64 padding_top,
46 tensorflow::int64 row_stride, tensorflow::int64 col_stride,
Druntime_conv2d_impl.h33 int64 row_stride, int64 col_stride, int64 padding_top, in EigenConvImpl() argument
75 .extract_image_patches(kernel_cols, kernel_rows, col_stride, in EigenConvImpl()
/external/mesa3d/src/mesa/drivers/dri/radeon/
Dradeon_maos_vbtmp.h47 GLuint tc0_stride, tc1_stride, col_stride, spec_stride, fog_stride; in TAG() local
127 col_stride = VB->AttribPtr[_TNL_ATTRIB_COLOR0]->stride; in TAG()
130 col_stride = 0; in TAG()
166 STRIDE_4F(col, start * col_stride); in TAG()
200 STRIDE_4F(col, col_stride); in TAG()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h99 const DenseIndex col_stride = 1,
152 static_cast<float>(col_stride));
156 out_width = numext::ceil(InputCols / static_cast<float>(col_stride));
218 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Deigen_spatial_convolutions.h96 const DenseIndex col_stride = 1,
149 static_cast<float>(col_stride));
153 out_width = numext::ceil(InputCols / static_cast<float>(col_stride));
215 .extract_image_patches(kernelRows, kernelCols, row_stride, col_stride,
/external/tensorflow/tensorflow/contrib/fused_conv/kernels/
Dfused_conv2d_bias_activation_op.h41 const Tensor& filter, int32 row_stride, int32 col_stride,
55 const Tensor& filter, int32 row_stride, int32 col_stride,

12