Home
last modified time | relevance | path

Searched refs:stride_cols (Results 1 – 25 of 28) sorted by relevance

12

/external/tensorflow/tensorflow/core/kernels/
Ddilation_ops.cc66 int* stride_rows, int* stride_cols, int* rate_rows, in ParseSizes() argument
82 *stride_cols = strides[2]; in ParseSizes()
110 context, GetWindowedOutputSize(input_cols, filter_cols_eff, *stride_cols, in ParseSizes()
126 int stride_rows = 0, stride_cols = 0; in Compute() local
130 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
151 filter.tensor<T, 3>(), stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
166 int stride_cols, int rate_rows, int rate_cols, int pad_top, in operator ()()
185 int w_beg = w_out * stride_cols - pad_left; in operator ()()
225 int stride_rows = 0, stride_cols = 0; in Compute() local
229 ParseSizes(context, strides_, rates_, padding_, &stride_rows, &stride_cols, in Compute()
[all …]
Dconv_3d.h47 int stride_rows, int stride_cols,
50 input, filter, stride_planes, stride_rows, stride_cols, padding);
60 int stride_planes, int stride_rows, int stride_cols) {
67 stride_cols, stride_rows, stride_planes);
77 int stride_planes, int stride_rows, int stride_cols) {
84 stride_cols, stride_rows, stride_planes);
Ddilation_ops_gpu.cu.cc44 int stride_cols, int rate_rows, int rate_cols, in DilationKernel() argument
55 int w_beg = w_out * stride_cols - pad_left; in DilationKernel()
83 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropInputKernel() argument
94 int w_beg = w_out * stride_cols - pad_left; in DilationBackpropInputKernel()
132 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropFilterKernel() argument
143 int w_beg = w_out * stride_cols - pad_left; in DilationBackpropFilterKernel()
183 int stride_cols, int rate_rows, int rate_cols, int pad_top, in operator ()()
203 output_cols, stride_rows, stride_cols, rate_rows, rate_cols, pad_top, in operator ()()
213 int stride_rows, int stride_cols, int rate_rows, in operator ()()
245 stride_rows, stride_cols, rate_rows, rate_cols, pad_top, pad_left, in operator ()()
[all …]
Dconv_ops_using_gemm.cc92 int stride_cols, Padding padding, T3* output_data, in operator ()() argument
112 ((output_width - 1) * stride_cols + filter_width - input_width + 1) / in operator ()()
119 ((output_width - 1) * stride_cols + filter_width - input_width) / 2; in operator ()()
152 const int in_x_origin = (out_x * stride_cols) - filter_left_offset; in operator ()()
215 int stride_cols, Padding padding, T3* output_data, in operator ()() argument
239 stride_cols == 1) { in operator ()()
273 ((output_width - 1) * stride_cols + filter_width - input_width + 1) / in operator ()()
280 ((output_width - 1) * stride_cols + filter_width - input_width) / 2; in operator ()()
340 const int in_x_origin = (out_x * stride_cols) - filter_left_offset; in operator ()()
518 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
[all …]
Dextract_image_patches_op.h30 int stride_cols, int rate_rows, int rate_cols, in operator()
40 .extract_image_patches(patch_cols, patch_rows, stride_cols, in operator()
46 .extract_image_patches(patch_cols, patch_rows, stride_cols, in operator()
Dextract_volume_patches_op.h31 int stride_planes, int stride_rows, int stride_cols, in operator()
41 stride_cols, stride_rows, stride_planes, in operator()
48 stride_cols, stride_rows, stride_planes, in operator()
Ddilation_ops.h31 int stride_cols, int rate_rows, int rate_cols, int pad_top,
44 int stride_rows, int stride_cols, int rate_rows,
58 int stride_rows, int stride_cols, int rate_rows,
Dextract_image_patches_op.cc78 const int stride_cols = strides_[2]; in Compute() local
92 GetWindowedOutputSize(in_cols, ksize_cols_eff, stride_cols, in Compute()
109 ksize_cols, stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
142 int patch_rows, int patch_cols, int stride_rows, int stride_cols, \
Dextract_volume_patches_op.cc87 const int stride_cols = strides_[3]; in Compute() local
118 GetWindowedOutputSize(in_cols, ksize_cols, stride_cols, in Compute()
136 ksize_rows, ksize_cols, stride_planes, stride_rows, stride_cols, in Compute()
172 int stride_planes, int stride_rows, int stride_cols, \
Dmkl_conv_ops.h77 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder() local
78 *strides = {stride_rows, stride_cols}; in GetStridesInMklOrder()
82 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder() local
83 *strides = {stride_planes, stride_rows, stride_cols}; in GetStridesInMklOrder()
353 int stride_planes, stride_rows, stride_cols; variable
358 stride_cols = strides[1];
365 stride_cols = strides[2];
411 input_cols, filter_cols, dilation_cols, stride_cols,
421 input_cols, filter_cols, stride_cols,
Dconv_ops.cc175 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
179 !CanUseDeepConv2D(stride_rows, stride_cols, filter_rows, filter_cols, in Run()
215 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run() argument
230 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
244 desc.v = stride_cols; in Run()
394 const int stride_cols = GetTensorDim(params.strides, params.data_format, 'W'); in ComputeConv2DDimension() local
414 input_cols, filter_cols, dilation_cols, stride_cols, params.padding, in ComputeConv2DDimension()
426 dimensions->stride_cols = stride_cols; in ComputeConv2DDimension()
481 << ", stride_cols = " << dimensions.stride_cols in Compute()
500 dimensions.stride_cols, output, params_.data_format)) { in Compute()
[all …]
Dconv_ops_fused_image_transform.cc263 int filter_count, int stride_rows, int stride_cols, in operator ()() argument
297 ((output_width - 1) * stride_cols + filter_width - padded_width + 1) / in operator ()()
304 ((output_width - 1) * stride_cols + filter_width - padded_width) / 2; in operator ()()
368 (((output_width - 1) * stride_cols) - filter_left_offset) + in operator ()()
542 const int in_x_origin = (out_x * stride_cols) - filter_left_offset; in operator ()()
823 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute() local
830 GetWindowedOutputSize(padded_cols, filter_cols, stride_cols, in Compute()
850 << ", stride_cols = " << stride_cols in Compute()
860 filter_cols, out_depth, stride_rows, stride_cols, padding_, in Compute()
Deigen_backward_spatial_convolutions_test.cc1810 const int stride_cols = 3; in TEST() local
1814 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols); in TEST()
1831 stride_planes, stride_rows, stride_cols); in TEST()
1855 output_k >= 0 && output_k / stride_cols < output_cols && in TEST()
1858 output_k % stride_cols == 0) { in TEST()
1862 output_k / stride_cols, b); in TEST()
1889 const int stride_cols = 3; in TEST() local
1893 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols); in TEST()
1910 stride_planes, stride_rows, stride_cols); in TEST()
1934 output_k >= 0 && output_k / stride_cols < output_cols && in TEST()
[all …]
Ddeep_conv2d.h100 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows,
Dconv_ops_fused_impl.h329 dimensions.stride_rows, dimensions.stride_cols,
619 0, (out_cols - 1) * dimensions.stride_cols +
699 .set_horizontal_filter_stride(dimensions.stride_cols)
765 dimensions.stride_cols}}, // stride_cols
920 << ", stride_cols = " << dimensions.stride_cols
Dconv_ops.h93 int stride_cols;
Dconv_grad_filter_ops.cc520 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
525 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
Ddeep_conv2d.cc97 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows, in CanUseDeepConv2D() argument
102 if (stride_rows > 1 || stride_cols > 1 || filter_rows != 3 || in CanUseDeepConv2D()
Dconv_grad_input_ops.cc657 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
662 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
/external/tensorflow/tensorflow/python/kernel_tests/
Dconv_ops_test.py1640 stride_cols, padding, test_input, data_format, argument
1647 output_cols = (input_cols - filter_cols + stride_cols) // stride_cols
1650 output_cols = (input_cols + stride_cols - 1) // stride_cols
1656 stride_cols) // stride_cols
1676 strides = [1, stride_rows, stride_cols, 1]
1728 stride_cols=1,
1746 stride_cols=1,
1764 stride_cols=2,
1782 stride_cols=2,
1800 stride_cols=3,
[all …]
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc465 const int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in Conv2DShapeImpl() local
531 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
582 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
593 stride_cols = strides[4]; in Conv3DShape()
600 stride_cols = strides[3]; in Conv3DShape()
631 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, -1, in Conv3DShape()
665 int32 stride_cols; in DepthwiseConv2DNativeShape() local
673 stride_cols = strides[3]; in DepthwiseConv2DNativeShape()
676 stride_cols = strides[2]; in DepthwiseConv2DNativeShape()
706 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols)); in DepthwiseConv2DNativeShape()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dmultithreaded_conv.h91 int stride_rows, int stride_cols, int pad_width, in operator()
95 stride_rows == 1 && stride_cols == 1); in operator()
129 Eigen::SpatialConvolution(input, filter, stride_cols, stride_rows, in operator()
/external/tensorflow/tensorflow/core/api_def/base_api/
Dapi_def_ExtractVolumePatches.pbtxt29 `input`. Must be: `[1, stride_planes, stride_rows, stride_cols, 1]`.
Dapi_def_ExtractImagePatches.pbtxt28 the images. Must be: `[1, stride_rows, stride_cols, 1]`.
/external/tensorflow/tensorflow/core/ops/
Dnn_ops.cc409 int32 stride_cols = strides[2]; in CommonFusedConvCalculations() local
428 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols)); in CommonFusedConvCalculations()
912 int32 stride_cols = strides[2]; in __anon3e672dd81d02() local
952 in_cols, filter_cols_eff, stride_cols, padding, &output_cols, in __anon3e672dd81d02()

12