/external/tensorflow/tensorflow/core/kernels/ |
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 | 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_gpu.cu.cc | 43 int output_cols, int stride_rows, in DilationKernel() argument 54 int h_beg = h_out * stride_rows - pad_top; in DilationKernel() 83 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationBackpropInputKernel() argument 93 int h_beg = h_out * stride_rows - pad_top; in DilationBackpropInputKernel() 132 int output_cols, int stride_rows, int stride_cols, int rate_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_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 …]
|
D | conv_ops_using_gemm.cc | 91 int filter_width, int filter_count, int stride_rows, in operator ()() argument 114 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()() 121 ((output_height - 1) * stride_rows + filter_height - input_height) / in operator ()() 153 const int in_y_origin = (out_y * stride_rows) - filter_top_offset; in operator ()() 214 int filter_width, int filter_count, int stride_rows, in operator ()() argument 238 if (filter_height == 1 && filter_width == 1 && stride_rows == 1 && in operator ()() 275 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()() 282 ((output_height - 1) * stride_rows + filter_height - input_height) / in operator ()() 339 const int in_y_origin = (out_y * stride_rows) - filter_top_offset; in operator ()() 517 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local [all …]
|
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 | 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 | extract_image_patches_op.cc | 77 const int stride_rows = strides_[1]; in Compute() local 89 GetWindowedOutputSize(in_rows, ksize_rows_eff, stride_rows, 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, \
|
D | extract_volume_patches_op.cc | 86 const int stride_rows = strides_[2]; in Compute() local 115 GetWindowedOutputSize(in_rows, ksize_rows, stride_rows, 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, \
|
D | mkl_conv_ops.h | 76 int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in GetStridesInMklOrder() local 78 *strides = {stride_rows, stride_cols}; in GetStridesInMklOrder() 81 int stride_rows = GetTensorDim(strides_, data_format_, '1'); in GetStridesInMklOrder() local 83 *strides = {stride_planes, stride_rows, stride_cols}; in GetStridesInMklOrder() 353 int stride_planes, stride_rows, stride_cols; variable 357 stride_rows = strides[0]; 364 stride_rows = strides[1]; 407 input_rows, filter_rows, dilation_rows, stride_rows, 418 input_rows, filter_rows, stride_rows,
|
D | conv_ops.cc | 175 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 243 desc.u = stride_rows; in Run() 393 const int stride_rows = GetTensorDim(params.strides, params.data_format, 'H'); in ComputeConv2DDimension() local 411 input_rows, filter_rows, dilation_rows, stride_rows, params.padding, in ComputeConv2DDimension() 425 dimensions->stride_rows = stride_rows; in ComputeConv2DDimension() 480 << ", stride_rows = " << dimensions.stride_rows in Compute() 499 dimensions.dilation_cols, dimensions.stride_rows, in Compute() [all …]
|
D | conv_ops_fused_image_transform.cc | 263 int filter_count, int stride_rows, int stride_cols, in operator ()() argument 299 filter_top_offset = ((output_height - 1) * stride_rows + filter_height - in operator ()() 306 ((output_height - 1) * stride_rows + filter_height - padded_height) / in operator ()() 431 ((output_height * stride_rows) - filter_top_offset) + filter_height; in operator ()() 433 const int in_y_origin = (out_y * stride_rows) - filter_top_offset; in operator ()() 822 const int stride_rows = GetTensorDim(strides_, FORMAT_NHWC, 'H'); in Compute() local 827 GetWindowedOutputSize(padded_rows, filter_rows, stride_rows, in Compute() 849 << ", stride_rows = " << stride_rows in Compute() 860 filter_cols, out_depth, stride_rows, stride_cols, padding_, in Compute()
|
D | eigen_backward_spatial_convolutions_test.cc | 1811 const int stride_rows = 1; in TEST() local 1813 const int output_rows = ceil_div(input_rows - patch_rows + 1, stride_rows); in TEST() 1831 stride_planes, stride_rows, stride_cols); in TEST() 1854 output_j >= 0 && output_j / stride_rows < output_rows && in TEST() 1857 output_j % stride_rows == 0 && in TEST() 1861 output_j / stride_rows, in TEST() 1890 const int stride_rows = 1; in TEST() local 1892 const int output_rows = ceil_div(input_rows - patch_rows + 1, stride_rows); in TEST() 1910 stride_planes, stride_rows, stride_cols); in TEST() 1933 output_j >= 0 && output_j / stride_rows < output_rows && in TEST() [all …]
|
D | deep_conv2d.h | 100 bool CanUseDeepConv2D(int stride_rows, int stride_cols, int filter_rows,
|
D | conv_ops_fused_impl.h | 329 dimensions.stride_rows, dimensions.stride_cols, 616 0, (out_rows - 1) * dimensions.stride_rows + 698 .set_vertical_filter_stride(dimensions.stride_rows) 764 {{dimensions.stride_rows, // stride_rows 919 << ", stride_rows = " << dimensions.stride_rows
|
D | conv_ops.h | 92 int stride_rows;
|
D | conv_grad_filter_ops.cc | 519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local 525 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()
|
D | conv_grad_input_ops.cc | 656 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); in Compute() local 662 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
|
/external/tensorflow/tensorflow/python/kernel_tests/ |
D | conv_ops_test.py | 1639 filter_cols, in_depth, out_depth, stride_rows, argument 1646 output_rows = (input_rows - filter_rows + stride_rows) // stride_rows 1649 output_rows = (input_rows + stride_rows - 1) // stride_rows 1654 stride_rows) // stride_rows 1676 strides = [1, stride_rows, stride_cols, 1] 1727 stride_rows=1, 1745 stride_rows=1, 1763 stride_rows=2, 1781 stride_rows=2, 1799 stride_rows=3, [all …]
|
/external/tensorflow/tensorflow/core/framework/ |
D | common_shape_fns.cc | 464 const int32 stride_rows = GetTensorDim(strides, data_format, 'H'); in Conv2DShapeImpl() local 528 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl() 582 int32 stride_planes, stride_rows, stride_cols; in Conv3DShape() local 592 stride_rows = strides[3]; in Conv3DShape() 599 stride_rows = strides[2]; in Conv3DShape() 628 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, -1, in Conv3DShape() 664 int32 stride_rows; in DepthwiseConv2DNativeShape() local 672 stride_rows = strides[2]; in DepthwiseConv2DNativeShape() 675 stride_rows = strides[1]; in DepthwiseConv2DNativeShape() 704 c, in_rows_dim, filter_rows_dim, stride_rows, padding, &output_rows)); in DepthwiseConv2DNativeShape() [all …]
|
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
D | multithreaded_conv.h | 91 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/ |
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 | 408 int32 stride_rows = strides[1]; in CommonFusedConvCalculations() local 426 c, in_rows_dim, filter_rows_dim, stride_rows, padding, &output_rows)); in CommonFusedConvCalculations() 911 int32 stride_rows = strides[1]; in __anon3e672dd81d02() local 949 in_rows, filter_rows_eff, stride_rows, padding, &output_rows, in __anon3e672dd81d02()
|