| /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
81 *stride_rows = strides[1]; in ParseSizes()
107 context, GetWindowedOutputSize(input_rows, filter_rows_eff, *stride_rows, 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()
165 typename TTypes<T, 3>::ConstTensor filter, int stride_rows, in operator ()()
183 int h_beg = h_out * stride_rows - pad_top; 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
52 int h_beg = h_out * stride_rows - pad_top; in DilationKernel()
81 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropInputKernel() argument
92 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropInputKernel()
131 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropFilterKernel() argument
142 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropFilterKernel()
182 typename TTypes<T, 3>::ConstTensor filter, int stride_rows, 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 | 93 int filter_width, int filter_count, int stride_rows, in operator ()() argument
116 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()()
123 ((output_height - 1) * stride_rows + filter_height - input_height) / in operator ()()
155 const int in_y_origin = (out_y * stride_rows) - filter_top_offset; in operator ()()
216 int filter_width, int filter_count, int stride_rows, in operator ()() argument
240 if (filter_height == 1 && filter_width == 1 && stride_rows == 1 && in operator ()()
277 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()()
284 ((output_height - 1) * stride_rows + filter_height - input_height) / in operator ()()
341 const int in_y_origin = (out_y * stride_rows) - filter_top_offset; in operator ()()
519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local
[all …]
|
| D | eigen_backward_cuboid_convolutions_test.cc | 711 const int stride_rows = 1; in TEST() local
713 const int output_rows = ceil_div(input_rows - patch_rows + 1, stride_rows); in TEST()
731 stride_planes, stride_rows, stride_cols); in TEST()
754 output_j >= 0 && output_j / stride_rows < output_rows && in TEST()
757 output_j % stride_rows == 0 && in TEST()
761 output_j / stride_rows, in TEST()
790 const int stride_rows = 1; in TEST() local
792 const int output_rows = ceil_div(input_rows - patch_rows + 1, stride_rows); in TEST()
810 stride_planes, stride_rows, stride_cols); in TEST()
833 output_j >= 0 && output_j / stride_rows < output_rows && in TEST()
[all …]
|
| D | dilation_ops.h | 30 typename TTypes<T, 3>::ConstTensor filter, int stride_rows,
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
300 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()()
307 ((output_height - 1) * stride_rows + filter_height - padded_height) / in operator ()()
432 ((output_height * stride_rows) - filter_top_offset) + filter_height; in operator ()()
434 const int in_y_origin = (out_y * stride_rows) - filter_top_offset; in operator ()()
819 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute() local
824 GetWindowedOutputSize(padded_rows, filter_rows, stride_rows, in Compute()
846 << ", stride_rows = " << stride_rows 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
310 desc.u = stride_rows; in Run()
460 const int stride_rows = GetTensorDim(params.strides, params.data_format, 'H'); in ComputeConv2DDimension() local
478 input_rows, filter_rows, dilation_rows, stride_rows, params.padding, in ComputeConv2DDimension()
492 dimensions->stride_rows = stride_rows; in ComputeConv2DDimension()
546 << ", stride_rows = " << dimensions.stride_rows in Compute()
565 dimensions.dilation_cols, dimensions.stride_rows, in Compute()
[all …]
|
| D | deep_conv2d.h | 100 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows,
|
| D | conv_ops.h | 92 int stride_rows;
|
| D | conv_ops_fused_impl.h | 247 dimensions.stride_rows, dimensions.stride_cols,
592 .set_vertical_filter_stride(dimensions.stride_rows)
657 {{dimensions.stride_rows, // stride_rows
785 << ", stride_rows = " << dimensions.stride_rows
|
| D | conv_grad_input_ops.h | 441 const int stride_rows = GetTensorDim(strides_, data_format_, 'H');
450 << " strides: [" << stride_rows << ", " << stride_cols << "]"
455 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_,
|
| D | conv_grad_filter_ops.cc | 349 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 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 | 29 int patch_rows, int patch_cols, int stride_rows, in operator()
41 stride_rows, rate_cols, rate_rows, padding) in operator()
47 stride_rows, rate_cols, rate_rows, padding) 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 | 89 const int stride_rows = strides_[2]; in Compute() local
118 GetWindowedOutputSize(in_rows, ksize_rows, stride_rows, 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 | 80 const int stride_rows = strides_[1]; in Compute() local
92 GetWindowedOutputSize(in_rows, ksize_rows_eff, stride_rows, 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 | 82 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder() local
84 *strides = {stride_rows, stride_cols}; in GetStridesInMklOrder()
87 int stride_rows = GetTensorDim(strides_, data_format_, '1'); in GetStridesInMklOrder() local
89 *strides = {stride_planes, stride_rows, stride_cols}; in GetStridesInMklOrder()
359 int stride_planes, stride_rows, stride_cols; variable
363 stride_rows = strides[0];
370 stride_rows = strides[1];
413 input_rows, filter_rows, dilation_rows, stride_rows,
426 input_rows, filter_rows, dilation_rows, stride_rows,
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | conv_ops_test.py | 864 stride_rows=1,
886 stride_rows=1,
1858 stride_rows, argument
1871 output_rows = (input_rows - filter_rows + stride_rows) // stride_rows
1874 output_rows = (input_rows + stride_rows - 1) // stride_rows
1879 stride_rows) // stride_rows
1901 strides = [1, stride_rows, stride_cols, 1]
1952 stride_rows=1,
1970 stride_rows=1,
1988 stride_rows=2,
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | multithreaded_conv.h | 90 int filter_width, int filter_count, int stride_rows, 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 | 631 const int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in Conv2DShapeImpl() local
711 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl()
765 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local
775 stride_rows = strides[3]; in Conv3DShape()
782 stride_rows = strides[2]; in Conv3DShape()
831 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, -1, in Conv3DShape()
958 int32 stride_rows; in DepthwiseConv2DNativeShapeImpl() local
968 stride_rows = strides[2]; in DepthwiseConv2DNativeShapeImpl()
973 stride_rows = strides[1]; in DepthwiseConv2DNativeShapeImpl()
1025 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, 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 | 488 int32 stride_rows = strides[1]; in CommonFusedConvCalculations() local
506 c, in_rows_dim, filter_rows_dim, stride_rows, padding, &output_rows)); in CommonFusedConvCalculations()
1005 int32 stride_rows = strides[1]; in __anon4650176f1702() local
1043 in_rows, filter_rows_eff, stride_rows, padding, &output_rows, in __anon4650176f1702()
|