Home
last modified time | relevance | path

Searched refs:col_dilation (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h32 int row_dilation, int col_dilation,
41 input, filter, col_stride, row_stride, padding, col_dilation,
52 int col_stride, int row_dilation, int col_dilation, in operator()
56 row_dilation, col_dilation, padding, output_kernel); in operator()
61 int col_stride, int row_dilation, int col_dilation, in operator()
67 col_dilation, Eigen::PaddingType::PADDING_VALID, output_kernel, in operator()
79 int col_dilation, const Eigen::PaddingType& padding,
83 col_stride, row_stride, padding, col_dilation,
92 int col_dilation, int padding_top, int padding_bottom,
98 Eigen::PaddingType::PADDING_VALID, col_dilation, row_dilation,
[all …]
Dconv_grad_ops.h176 int row_dilation, int col_dilation, int row_stride,
186 int row_dilation, int col_dilation, int row_stride,
197 int col_dilation, int row_stride, int col_stride,
207 int row_dilation, int col_dilation, int row_stride,
Dconv_ops.cc82 int row_dilation, int col_dilation, const Padding& padding, in operator ()()
110 col_dilation == 1 && padding == VALID) { in operator ()()
128 row_dilation, col_dilation, static_cast<int>(explicit_paddings[2]), in operator ()()
136 row_dilation, col_dilation, BrainPadding2EigenPadding(padding)); in operator ()()
147 int col_dilation, int row_stride, int col_stride, in operator ()()
174 row_dilation, col_dilation, padding, in operator ()()
184 int col_dilation, int row_stride, int col_stride, in operator ()()
212 ctx, input, filter, row_stride, col_stride, row_dilation, col_dilation, in operator ()()
649 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
679 row_dilation == 1 && col_dilation == 1 && row_stride == 1 && in operator ()()
[all …]
Dconv_ops.h38 int col_dilation, int row_stride, int col_stride,
49 int col_dilation, int row_stride, int col_stride,
Dconv_grad_input_ops.h113 int row_dilation, int col_dilation, int row_stride, in operator()
125 dilations[input_w] = col_dilation; in operator()
157 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator()
190 col_stride, row_stride, col_dilation, row_dilation); in operator()
197 col_stride, row_stride, col_dilation, row_dilation, padding_top, in operator()
208 int row_dilation, int col_dilation, int row_stride,
214 row_dilation, col_dilation, row_stride, col_stride, padding,
Dconv_grad_filter_ops.cc110 int row_dilation, int col_dilation, int row_stride, in operator ()()
116 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
148 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
167 filter_backprop_t.dimension(0), col_stride, row_stride, col_dilation, in operator ()()
185 filter_backprop_t.dimension(0), col_stride, row_stride, col_dilation, in operator ()()
653 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
662 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
694 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
Dconv_grad_input_ops.cc53 int row_dilation, int col_dilation, int row_stride, in operator ()()
59 row_dilation, col_dilation, row_stride, col_stride, padding, in operator ()()
69 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
83 dilations[input_w] = col_dilation; in operator ()()
112 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
Dconv_ops_fused_impl.h96 int row_dilation, int col_dilation, // in LaunchFusedConv2DWithOutputKernel() argument
102 col_dilation_(col_dilation), in LaunchFusedConv2DWithOutputKernel()