| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_3d.h | 47 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);
|
| D | dilation_ops.cc | 66 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 …]
|
| D | dilation_ops_gpu.cu.cc | 42 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationKernel() argument
53 int w_beg = w_out * stride_cols - pad_left; in DilationKernel()
82 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, in DilationBackpropInputKernel() argument
93 int w_beg = w_out * stride_cols - pad_left; in DilationBackpropInputKernel()
132 int stride_cols, int rate_rows, int rate_cols, int pad_top, int pad_left, 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_rows, output_cols, stride_rows, stride_cols, rate_rows, 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 …]
|
| D | conv_ops_using_gemm.cc | 94 int stride_cols, Padding padding, T3* output_data, in operator ()() argument
114 ((output_width - 1) * stride_cols + filter_width - input_width + 1) / in operator ()()
121 ((output_width - 1) * stride_cols + filter_width - input_width) / 2; in operator ()()
154 const int in_x_origin = (out_x * stride_cols) - filter_left_offset; in operator ()()
217 int stride_cols, Padding padding, T3* output_data, in operator ()() argument
241 stride_cols == 1) { in operator ()()
275 ((output_width - 1) * stride_cols + filter_width - input_width + 1) / in operator ()()
282 ((output_width - 1) * stride_cols + filter_width - input_width) / 2; in operator ()()
342 const int in_x_origin = (out_x * stride_cols) - filter_left_offset; in operator ()()
520 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
[all …]
|
| D | eigen_backward_cuboid_convolutions_test.cc | 710 const int stride_cols = 3; in TEST() local
714 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols); in TEST()
731 stride_planes, stride_rows, stride_cols); in TEST()
755 output_k >= 0 && output_k / stride_cols < output_cols && in TEST()
758 output_k % stride_cols == 0) { in TEST()
762 output_k / stride_cols, b); in TEST()
789 const int stride_cols = 3; in TEST() local
793 const int output_cols = ceil_div(input_cols - patch_cols + 1, stride_cols); in TEST()
810 stride_planes, stride_rows, stride_cols); in TEST()
834 output_k >= 0 && output_k / stride_cols < output_cols && in TEST()
[all …]
|
| D | dilation_ops.h | 31 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,
|
| D | conv_ops_fused_image_transform.cc | 264 int filter_count, int stride_rows, int stride_cols, in operator ()() argument
298 ((output_width - 1) * stride_cols + filter_width - padded_width + 1) / in operator ()()
305 ((output_width - 1) * stride_cols + filter_width - padded_width) / 2; in operator ()()
369 (((output_width - 1) * stride_cols) - filter_left_offset) + in operator ()()
543 const int in_x_origin = (out_x * stride_cols) - filter_left_offset; in operator ()()
820 const int stride_cols = GetTensorDim(strides_, FORMAT_NHWC, 'W'); in Compute() local
827 GetWindowedOutputSize(padded_cols, filter_cols, stride_cols, in Compute()
847 << ", stride_cols = " << stride_cols in Compute()
857 filter_cols, out_depth, stride_rows, stride_cols, padding_, in Compute()
|
| D | conv_ops.cc | 242 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
246 !CanUseDeepConv2D(stride_rows, stride_cols, filter_rows, filter_cols, in Run()
282 int out_cols, int out_depth, int stride_rows, int stride_cols, in Run() argument
297 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
311 desc.v = stride_cols; in Run()
461 const int stride_cols = GetTensorDim(params.strides, params.data_format, 'W'); in ComputeConv2DDimension() local
481 input_cols, filter_cols, dilation_cols, stride_cols, params.padding, in ComputeConv2DDimension()
493 dimensions->stride_cols = stride_cols; in ComputeConv2DDimension()
547 << ", stride_cols = " << dimensions.stride_cols in Compute()
566 dimensions.stride_cols, output, params_.data_format)) { in Compute()
[all …]
|
| D | deep_conv2d.h | 100 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows,
|
| D | conv_ops.h | 93 int stride_cols;
|
| D | conv_ops_fused_impl.h | 247 dimensions.stride_rows, dimensions.stride_cols,
593 .set_horizontal_filter_stride(dimensions.stride_cols)
658 dimensions.stride_cols}}, // stride_cols
786 << ", stride_cols = " << dimensions.stride_cols
|
| D | conv_grad_input_ops.h | 442 const int stride_cols = GetTensorDim(strides_, data_format_, 'W');
450 << " strides: [" << stride_rows << ", " << stride_cols << "]"
455 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_,
|
| D | conv_grad_filter_ops.cc | 350 const int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in Compute() local
358 << " strides: [" << stride_rows << ", " << stride_cols << "]" in Compute()
362 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
|
| D | deep_conv2d.cc | 97 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()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | extract_image_patches_op.h | 30 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()
|
| D | extract_volume_patches_op.h | 31 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()
|
| D | extract_volume_patches_op.cc | 90 const int stride_cols = strides_[3]; in Compute() local
121 GetWindowedOutputSize(in_cols, ksize_cols, stride_cols, in Compute()
139 ksize_rows, ksize_cols, stride_planes, stride_rows, stride_cols, in Compute()
175 int stride_planes, int stride_rows, int stride_cols, \
|
| D | extract_image_patches_op.cc | 81 const int stride_cols = strides_[2]; in Compute() local
95 GetWindowedOutputSize(in_cols, ksize_cols_eff, stride_cols, in Compute()
112 ksize_cols, stride_rows, stride_cols, rate_rows, rate_cols, in Compute()
146 int patch_rows, int patch_cols, int stride_rows, int stride_cols, \
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_conv_ops.h | 83 int stride_cols = GetTensorDim(strides_, data_format_, 'W'); in GetStridesInMklOrder() local
84 *strides = {stride_rows, stride_cols}; in GetStridesInMklOrder()
88 int stride_cols = GetTensorDim(strides_, data_format_, '2'); in GetStridesInMklOrder() local
89 *strides = {stride_planes, stride_rows, stride_cols}; in GetStridesInMklOrder()
359 int stride_planes, stride_rows, stride_cols; variable
364 stride_cols = strides[1];
371 stride_cols = strides[2];
417 input_cols, filter_cols, dilation_cols, stride_cols,
430 input_cols, filter_cols, dilation_cols, stride_cols,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_test.py | 865 stride_cols=1,
887 stride_cols=1,
1859 stride_cols, argument
1872 output_cols = (input_cols - filter_cols + stride_cols) // stride_cols
1875 output_cols = (input_cols + stride_cols - 1) // stride_cols
1881 stride_cols) // stride_cols
1901 strides = [1, stride_rows, stride_cols, 1]
1953 stride_cols=1,
1971 stride_cols=1,
1989 stride_cols=2,
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 91 int stride_cols, int pad_width, int pad_height, 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/framework/ |
| D | common_shape_fns.cc | 632 const int32 stride_cols = GetTensorDim(strides, data_format, 'W'); in Conv2DShapeImpl() local
714 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
765 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
776 stride_cols = strides[4]; in Conv3DShape()
783 stride_cols = strides[3]; in Conv3DShape()
834 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, -1, in Conv3DShape()
959 int32 stride_cols; in DepthwiseConv2DNativeShapeImpl() local
969 stride_cols = strides[3]; in DepthwiseConv2DNativeShapeImpl()
974 stride_cols = strides[2]; in DepthwiseConv2DNativeShapeImpl()
1028 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, in DepthwiseConv2DNativeShapeImpl()
[all …]
|
| /external/tensorflow/tensorflow/core/api_def/base_api/ |
| D | api_def_ExtractVolumePatches.pbtxt | 29 `input`. Must be: `[1, stride_planes, stride_rows, stride_cols, 1]`.
|
| D | api_def_ExtractImagePatches.pbtxt | 28 the images. Must be: `[1, stride_rows, stride_cols, 1]`.
|
| /external/tensorflow/tensorflow/core/ops/ |
| D | nn_ops.cc | 489 int32 stride_cols = strides[2]; in CommonFusedConvCalculations() local
508 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols)); in CommonFusedConvCalculations()
1006 int32 stride_cols = strides[2]; in __anonad10f3361702() local
1046 in_cols, filter_cols_eff, stride_cols, padding, &output_cols, in __anonad10f3361702()
|