Home
last modified time | relevance | path

Searched refs:stride_rows (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
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 …]
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.cc43 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 …]
Dconv_ops_using_gemm.cc91 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 …]
Dextract_image_patches_op.h29 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()
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.h30 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,
Dextract_image_patches_op.cc77 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, \
Dextract_volume_patches_op.cc86 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, \
Dmkl_conv_ops.h76 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,
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
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 …]
Dconv_ops_fused_image_transform.cc263 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()
Deigen_backward_spatial_convolutions_test.cc1811 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 …]
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,
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
Dconv_ops.h92 int stride_rows;
Dconv_grad_filter_ops.cc519 const int stride_rows = GetTensorDim(strides_, data_format_, 'H'); 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.cc656 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/
Dconv_ops_test.py1639 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/
Dcommon_shape_fns.cc464 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/
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.cc408 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()

12