Searched refs:dilation_rows (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_ops.cc | 174 int out_cols, int out_depth, int dilation_rows, in Run() argument
177 if (data_format != FORMAT_NHWC || dilation_rows != 1 || in Run()
216 int dilation_rows, int dilation_cols, Tensor* output, in Run() argument
229 int out_cols, int out_depth, int dilation_rows, in Run() argument
259 if (dilation_rows != 1 || dilation_cols != 1 || in Run()
395 const int dilation_rows = in ComputeConv2DDimension() local
411 input_rows, filter_rows, dilation_rows, stride_rows, params.padding, in ComputeConv2DDimension()
427 dimensions->dilation_rows = dilation_rows; in ComputeConv2DDimension()
482 << ", dilation_rows = " << dimensions.dilation_rows in Compute()
498 dimensions.out_cols, dimensions.out_depth, dimensions.dilation_rows, in Compute()
[all …]
|
| D | mkl_conv_ops.h | 354 int dilation_planes, dilation_rows, dilation_cols; variable
359 dilation_rows = dilations[0];
367 dilation_rows = dilations[1];
407 input_rows, filter_rows, dilation_rows, stride_rows,
|
| D | conv_ops_fused_impl.h | 330 dimensions.dilation_rows, dimensions.dilation_cols, params.padding);
617 (patch_rows - 1) * dimensions.dilation_rows + 1 - in_rows);
696 conv_desc.set_vertical_dilation_rate(dimensions.dilation_rows)
762 {{dimensions.dilation_rows, // dilation_rows
921 << ", dilation_rows = " << dimensions.dilation_rows
|
| D | conv_ops.h | 95 int dilation_rows;
|
| D | conv_grad_filter_ops.cc | 521 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H'); in Compute() local
525 dilation_rows, dilation_cols, stride_rows, stride_cols, padding_, in Compute()
|
| D | conv_grad_input_ops.cc | 658 const int dilation_rows = GetTensorDim(dilations_, data_format_, 'H'); 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 | 466 const int32 dilation_rows = GetTensorDim(dilations, data_format, 'H'); in Conv2DShapeImpl() local
528 c, input_spatial_dims[0], filter_rows_dim, dilation_rows, stride_rows, in Conv2DShapeImpl()
583 int32 dilation_planes, dilation_rows, dilation_cols; in Conv3DShape() local
596 dilation_rows = dilations[4]; in Conv3DShape()
603 dilation_rows = dilations[3]; in Conv3DShape()
628 c, in_rows_dim, filter_rows_dim, dilation_rows, stride_rows, padding, -1, in Conv3DShape()
|