Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/kernel_tests/
Dpooling_ops_test.py1115 col_stride, argument
1158 strides = [1, 1, row_stride, col_stride]
1164 strides = [1, row_stride, col_stride, 1]
1193 col_stride, argument
1236 strides = [1, 1, row_stride, col_stride]
1240 strides = [1, row_stride, col_stride, 1]
1272 col_stride=1,
1286 col_stride=1,
1300 col_stride=1,
1314 col_stride=2,
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h31 Filter filter, int row_stride, int col_stride,
41 input, filter, col_stride, row_stride, padding, col_dilation,
52 int col_stride, int row_dilation, int col_dilation, in operator()
55 SpatialConvolutionFunc(d, output, input, filter, row_stride, col_stride, in operator()
61 int col_stride, int row_dilation, int col_dilation, in operator()
66 d, output, input, filter, row_stride, col_stride, row_dilation, in operator()
78 int row_stride, int col_stride, int row_dilation,
83 col_stride, row_stride, padding, col_dilation,
91 int row_stride, int col_stride, int row_dilation,
97 input.cast<float>(), filter.cast<float>(), col_stride, row_stride,
[all …]
Dconv_grad_ops.h177 int col_stride, const Padding& padding,
187 int col_stride, const Padding& padding,
197 int col_dilation, int row_stride, int col_stride,
208 int col_stride, const Padding& padding,
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,
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()
Deigen_backward_spatial_convolutions.h103 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
155 0, ((outputCols - 1) * col_stride + kernelColsEff - inputCols) / 2);
161 const TensorIndex padding_right = inputCols - (outputCols - 1) * col_stride -
287 col_in_stride, row_stride, col_stride, padding_top,
294 col_in_stride, row_stride, col_stride,
403 const DenseIndex row_stride = 1, const DenseIndex col_stride = 1,
421 eigen_assert(!(col_stride > 1 && col_in_stride > 1));
470 0, (outputCols - 1) * col_stride + kernelColsEff - inputCols);
477 const TensorIndex expanded_out_cols = (outputCols - 1) * col_stride + 1;
592 col_stride, top_pad_rows,
[all …]
Dpooling_ops_common.h69 int col_stride; member
207 params.row_stride, params.col_stride, pt); in SpatialMaxPool()
243 const int32 col_stride = params.col_stride; in SpatialMaxPool() local
269 : (wpad - window_cols) / col_stride + 1; in SpatialMaxPool()
270 const int32 w_end = std::min(wpad / col_stride + 1, out_width); in SpatialMaxPool()
316 params.window_cols, params.row_stride, params.col_stride,
465 params.row_stride, params.col_stride, pt);
504 const int32 col_stride = params.col_stride;
530 : (wpad - window_cols) / col_stride + 1;
531 const int32 w_end = std::min(wpad / col_stride + 1, out_width);
[all …]
Davgpooling_op.cc204 params.row_stride, params.col_stride, pt); in Compute()
223 int window_cols, int row_stride, int col_stride, \
313 const int col_stride = stride_[2]; in Compute() local
328 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
337 col_stride, pad_rows, pad_cols](int64 start, int64 limit) { in Compute()
357 GetBroadcastSize(c, in_cols, window_cols, col_stride, in Compute()
573 const int col_stride = stride_[2]; in Compute() local
588 GetWindowedOutputSize(in_cols, window_cols, col_stride, in Compute()
601 col_stride, // stride_w in Compute()
Dconv_grad_input_ops.h114 int col_stride, const Padding& padding, in operator()
123 strides[input_w] = col_stride; in operator()
157 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator()
190 col_stride, row_stride, col_dilation, row_dilation); in operator()
197 col_stride, row_stride, col_dilation, row_dilation, padding_top, in operator()
209 int col_stride, const Padding& padding,
214 row_dilation, col_dilation, row_stride, col_stride, padding,
227 int col_stride, int pad_h, int pad_w,
240 int col_stride, int pad_h, int pad_w,
259 desc.v = col_stride;
Dconv_ops.cc81 const Tensor& filter, int row_stride, int col_stride, in operator ()()
88 col_stride == 1 && (padding == SAME || padding == VALID)) { in operator ()()
127 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
135 input.tensor<T, 4>(), filter.tensor<T, 4>(), row_stride, col_stride, in operator ()()
147 int col_dilation, int row_stride, int col_stride, in operator ()()
173 LaunchGeneric<CPUDevice, T>()(ctx, input, filter, row_stride, col_stride, in operator ()()
184 int col_dilation, int row_stride, int col_stride, in operator ()()
212 ctx, input, filter, row_stride, col_stride, row_dilation, col_dilation, in operator ()()
649 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
680 col_stride == 1 && data_format == FORMAT_NHWC && 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,
Dconv_grad_filter_ops.cc111 int col_stride, const Padding& padding, in operator ()()
120 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
148 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
167 filter_backprop_t.dimension(0), col_stride, row_stride, col_dilation, in operator ()()
185 filter_backprop_t.dimension(0), col_stride, row_stride, col_dilation, in operator ()()
199 int col_stride, int pad_h, int pad_w, in operator ()()
211 int input_cols, int row_stride, int col_stride, int pad_h, in operator ()()
231 desc.v = col_stride; in operator ()()
653 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
666 strides[GetTensorDimIndex(data_format, 'W')] = col_stride; in operator ()()
[all …]
Dops_util_test.cc40 int col_stride; member
81 pad_struct.input.col_stride, pad_struct.input.padding, &new_width, in VerifyGet2dOutputSizeBoundaries()
96 pad_struct.input.col_stride, pad_struct.input.padding, &new_width, in VerifyGet2dOutputSizeValues()
115 pad_struct.input.col_stride, pad_struct.input.padding, &new_width, in VerifyGet2dOutputVerboseSizeValues()
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,
/external/gemmlowp/internal/
Doutput_neon.h391 int col_stride = dst->cols_stride();
393 vst1q_s32(dst_ptr + i * col_stride + 0, src.buf.reg[2 * i + 0]);
394 vst1q_s32(dst_ptr + i * col_stride + 4, src.buf.reg[2 * i + 1]);
462 int col_stride = dst->cols_stride();
464 vst1q_s32(dst_ptr + i * col_stride, src.buf.reg[2 * i]);
465 vst1q_s32(dst_ptr + i * col_stride + 4, src.buf.reg[2 * i + 1]);
536 int col_stride = dst->cols_stride();
537 vst1q_lane_s32(dst_ptr + 0 * col_stride, src.buf.reg[0], 0);
538 vst1q_lane_s32(dst_ptr + 1 * col_stride, src.buf.reg[0], 1);
539 vst1q_lane_s32(dst_ptr + 2 * col_stride, src.buf.reg[0], 2);
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Druntime_single_threaded_conv2d.cc31 tensorflow::int64 col_stride, tensorflow::int64 padding_top, in __xla_cpu_runtime_EigenSingleThreadedConvF16() argument
39 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF16()
52 tensorflow::int64 row_stride, tensorflow::int64 col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32() argument
60 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_EigenSingleThreadedConvF32()
Druntime_conv2d.cc33 tensorflow::int64 row_stride, tensorflow::int64 col_stride, in __xla_cpu_runtime_EigenConvF32() argument
45 col_stride, padding_top, padding_bottom, padding_left, padding_right, in __xla_cpu_runtime_EigenConvF32()
57 tensorflow::int64 col_stride, tensorflow::int64 padding_top, in __xla_cpu_runtime_EigenConvF16() argument
69 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()
159 int64 row_stride, int64 col_stride, int64 padding_top, int64 padding_bottom, in __xla_cpu_runtime_MKLConvF32() argument
169 kernel_filters, output_rows, output_cols, row_stride, col_stride, in __xla_cpu_runtime_MKLConvF32()
176 col_stride, padding_top, padding_bottom, padding_left, in __xla_cpu_runtime_MKLConvF32()
Druntime_conv2d.h32 tensorflow::int64 col_stride, tensorflow::int64 padding_top,
46 tensorflow::int64 row_stride, tensorflow::int64 col_stride,
Druntime_single_threaded_conv2d.h32 tensorflow::int64 row_stride, tensorflow::int64 col_stride,
46 tensorflow::int64 col_stride, tensorflow::int64 padding_top,
Druntime_conv2d_impl.h39 Eigen::Index col_stride, Eigen::Index padding_top, in EigenConvImpl() argument
81 .extract_image_patches(kernel_cols, kernel_rows, col_stride, in EigenConvImpl()
/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/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/libopus/src/
Dmlp.c72 static void gemm_accum(float *out, const opus_int8 *weights, int rows, int cols, int col_stride, co… in gemm_accum() argument
78 out[i] += weights[j*col_stride + i]*x[j]; in gemm_accum()
/external/ruy/ruy/
Dmatrix.h197 int col_stride = layout.order() == Order::kRowMajor ? 1 : layout.stride(); in Offset() local
198 return row * row_stride + col * col_stride; in Offset()

12