| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d.h | 58 int row_dilation, int col_dilation, in SpatialConvolutionFunc() argument
64 col_dilation, row_dilation, output_kernel); in SpatialConvolutionFunc()
73 int col_stride, int row_dilation, int col_dilation, in operator()
77 row_dilation, col_dilation, padding, output_kernel); in operator()
87 int row_stride, int col_stride, int row_dilation,
93 row_dilation, output_kernel)
103 int row_stride, int col_stride, int row_dilation,
109 row_dilation);
119 int row_stride, int col_stride, int row_dilation,
125 row_dilation);
|
| D | conv_grad_ops.h | 178 int row_dilation, int col_dilation, int row_stride,
188 int row_dilation, int col_dilation, int row_stride,
198 const Tensor& input, const Tensor& filter, int row_dilation,
209 int row_dilation, int col_dilation, int row_stride,
|
| D | conv_ops.cc | 72 int row_dilation, int col_dilation, const Padding& padding, in operator ()()
98 filter.dim_size(1) == input.dim_size(2) && row_dilation == 1 && in operator ()()
116 row_dilation, col_dilation, BrainPadding2EigenPadding(padding)); in operator ()()
125 const Tensor& input, const Tensor& filter, int row_dilation, in operator ()()
146 row_dilation, col_dilation, padding, output, in operator ()()
580 const Tensor& input_param, const Tensor& filter, int row_dilation, in operator ()() argument
611 row_dilation == 1 && col_dilation == 1 && row_stride == 1 && in operator ()()
638 !is_grouped_convolution && row_dilation == 1 && in operator ()()
681 in_rows, patch_rows, row_dilation, row_stride, padding, &out_rows_check, in operator ()()
781 conv_desc.set_vertical_dilation_rate(row_dilation) in operator ()()
[all …]
|
| D | conv_ops.h | 37 const Tensor& input, const Tensor& filter, int row_dilation,
48 const Tensor& input, const Tensor& filter, int row_dilation,
|
| D | conv_grad_filter_ops.cc | 105 int row_dilation, int col_dilation, int row_stride, in operator ()()
545 const Tensor& out_backprop, const Tensor& input, int row_dilation, in operator ()() argument
554 dilations[GetTensorDimIndex(data_format, 'H')] = row_dilation; in operator ()()
582 row_dilation, row_stride, padding, &expected_out_rows, &padding_top, in operator ()()
|
| D | conv_grad_input_ops.cc | 111 int row_dilation, int col_dilation, int row_stride, in operator ()()
682 const Tensor& out_backprop, const Tensor& filter, int row_dilation, in operator ()() argument
696 dilations[input_h] = row_dilation; in operator ()()
721 row_dilation, row_stride, padding, &expected_out_rows, &padding_top, in operator ()()
|
| D | conv_ops_fused_impl.h | 249 int row_dilation, int col_dilation, // in LaunchFusedConv2DWithOutputKernel() argument
253 row_dilation_(row_dilation), in LaunchFusedConv2DWithOutputKernel()
|