Home
last modified time | relevance | path

Searched refs:dilation_cols (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_ops.cc242 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
245 dilation_cols != 1 || in Run()
283 int dilation_rows, int dilation_cols, Tensor* output, in Run() argument
297 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
326 if (dilation_rows != 1 || dilation_cols != 1 || in Run()
464 const int dilation_cols = in ComputeConv2DDimension() local
481 input_cols, filter_cols, dilation_cols, stride_cols, params.padding, in ComputeConv2DDimension()
495 dimensions->dilation_cols = dilation_cols; in ComputeConv2DDimension()
549 << ", dilation_cols = " << dimensions.dilation_cols in Compute()
565 dimensions.dilation_cols, dimensions.stride_rows, in Compute()
[all …]
Dconv_ops.h96 int dilation_cols;
Dconv_ops_fused_impl.h248 dimensions.dilation_rows, dimensions.dilation_cols, params.padding,
591 .set_horizontal_dilation_rate(dimensions.dilation_cols)
656 dimensions.dilation_cols}}, // dilation_cols
788 << ", dilation_cols = " << dimensions.dilation_cols
Dconv_grad_input_ops.h444 const int dilation_cols = GetTensorDim(dilations_, data_format_, 'W');
451 << " dilations: [" << dilation_rows << ", " << dilation_cols << "]";
455 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_,
Dconv_grad_filter_ops.cc352 const int dilation_cols = GetTensorDim(dilations_, data_format_, 'W'); in Compute() local
359 << " dilations: [" << dilation_rows << ", " << dilation_cols << "]"; in Compute()
362 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_conv_ops.h360 int dilation_planes, dilation_rows, dilation_cols; variable
366 dilation_cols = dilations[1];
374 dilation_cols = dilations[2];
417 input_cols, filter_cols, dilation_cols, stride_cols,
430 input_cols, filter_cols, dilation_cols, stride_cols,
/external/tensorflow/tensorflow/core/framework/
Dcommon_shape_fns.cc634 const int32 dilation_cols = GetTensorDim(dilations, data_format, 'W'); in Conv2DShapeImpl() local
714 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
766 int32 dilation_planes, dilation_rows, dilation_cols; in Conv3DShape() local
778 dilation_cols = dilations[3]; in Conv3DShape()
785 dilation_cols = dilations[2]; in Conv3DShape()
834 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, -1, in Conv3DShape()
961 int32 dilation_cols; in DepthwiseConv2DNativeShapeImpl() local
971 dilation_cols = dilations[3]; in DepthwiseConv2DNativeShapeImpl()
976 dilation_cols = dilations[2]; in DepthwiseConv2DNativeShapeImpl()
1028 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, in DepthwiseConv2DNativeShapeImpl()