Searched refs:dilation_cols (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 175 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
178 dilation_cols != 1 || in Run()
216 int dilation_rows, int dilation_cols, Tensor* output, in Run() argument
230 int dilation_cols, int stride_rows, int stride_cols, in Run() argument
259 if (dilation_rows != 1 || dilation_cols != 1 || in Run()
397 const int dilation_cols = in ComputeConv2DDimension() local
414 input_cols, filter_cols, dilation_cols, stride_cols, params.padding, in ComputeConv2DDimension()
428 dimensions->dilation_cols = dilation_cols; in ComputeConv2DDimension()
483 << ", dilation_cols = " << dimensions.dilation_cols in Compute()
499 dimensions.dilation_cols, dimensions.stride_rows, in Compute()
[all …]
|
| D | mkl_conv_ops.h | 354 int dilation_planes, dilation_rows, dilation_cols; variable
360 dilation_cols = dilations[1];
368 dilation_cols = dilations[2];
411 input_cols, filter_cols, dilation_cols, stride_cols,
|
| D | conv_ops_fused_impl.h | 330 dimensions.dilation_rows, dimensions.dilation_cols, params.padding);
620 (patch_cols - 1) * dimensions.dilation_cols + 1 - in_cols);
697 .set_horizontal_dilation_rate(dimensions.dilation_cols)
763 dimensions.dilation_cols}}, // dilation_cols
922 << ", dilation_cols = " << dimensions.dilation_cols
|
| D | conv_ops.h | 96 int dilation_cols;
|
| D | conv_grad_filter_ops.cc | 522 const int dilation_cols = GetTensorDim(dilations_, data_format_, 'W'); in Compute() local
525 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
|
| D | conv_grad_input_ops.cc | 659 const int dilation_cols = GetTensorDim(dilations_, data_format_, 'W'); in Compute() local
662 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | common_shape_fns.cc | 467 const int32 dilation_cols = GetTensorDim(dilations, data_format, 'W'); in Conv2DShapeImpl() local
531 c, input_spatial_dims[1], filter_cols_dim, dilation_cols, stride_cols, in Conv2DShapeImpl()
583 int32 dilation_planes, dilation_rows, dilation_cols; in Conv3DShape() local
595 dilation_cols = dilations[3]; in Conv3DShape()
602 dilation_cols = dilations[2]; in Conv3DShape()
631 c, in_cols_dim, filter_cols_dim, dilation_cols, stride_cols, padding, -1, in Conv3DShape()
|