| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 241 int out_cols, int out_depth, int dilation_rows, in Run() argument
244 if (data_format != FORMAT_NHWC || dilation_rows != 1 || in Run()
283 int dilation_rows, int dilation_cols, Tensor* output, in Run() argument
296 int out_cols, int out_depth, int dilation_rows, in Run() argument
326 if (dilation_rows != 1 || dilation_cols != 1 || in Run()
462 const int dilation_rows = in ComputeConv2DDimension() local
478 input_rows, filter_rows, dilation_rows, stride_rows, params.padding, in ComputeConv2DDimension()
494 dimensions->dilation_rows = dilation_rows; in ComputeConv2DDimension()
548 << ", dilation_rows = " << dimensions.dilation_rows in Compute()
564 dimensions.out_cols, dimensions.out_depth, dimensions.dilation_rows, in Compute()
[all …]
|
| D | conv_ops.h | 95 int dilation_rows;
|
| D | conv_ops_fused_impl.h | 248 dimensions.dilation_rows, dimensions.dilation_cols, params.padding,
590 conv_desc.set_vertical_dilation_rate(dimensions.dilation_rows)
655 {{dimensions.dilation_rows, // dilation_rows
787 << ", dilation_rows = " << dimensions.dilation_rows
|
| D | conv_grad_input_ops.h | 443 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H');
451 << " dilations: [" << dilation_rows << ", " << dilation_cols << "]";
455 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_,
|
| D | conv_grad_filter_ops.cc | 351 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H'); 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/ |
| D | mkl_conv_ops.h | 360 int dilation_planes, dilation_rows, dilation_cols; variable
365 dilation_rows = dilations[0];
373 dilation_rows = dilations[1];
413 input_rows, filter_rows, dilation_rows, stride_rows,
426 input_rows, filter_rows, dilation_rows, stride_rows,
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 633 const int32 dilation_rows = GetTensorDim(dilations, data_format, 'H'); in Conv2DShapeImpl() local
711 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl()
766 int32 dilation_planes, dilation_rows, dilation_cols; in Conv3DShape() local
779 dilation_rows = dilations[4]; in Conv3DShape()
786 dilation_rows = dilations[3]; in Conv3DShape()
831 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, -1, in Conv3DShape()
960 int32 dilation_rows; in DepthwiseConv2DNativeShapeImpl() local
970 dilation_rows = dilations[2]; in DepthwiseConv2DNativeShapeImpl()
975 dilation_rows = dilations[1]; in DepthwiseConv2DNativeShapeImpl()
1025 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, in DepthwiseConv2DNativeShapeImpl()
|