Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconv_2d.h58 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()
88 int col_dilation, const Eigen::PaddingType& padding,
92 col_stride, row_stride, padding, col_dilation,
104 int col_dilation) {
108 input_backward.dimension(1), col_stride, row_stride, col_dilation,
120 int col_dilation) {
124 kernel_backward.dimension(0), col_stride, row_stride, col_dilation,
Dconv_grad_ops.h178 int row_dilation, int col_dilation, int row_stride,
188 int row_dilation, int col_dilation, int row_stride,
199 int col_dilation, int row_stride, int col_stride,
209 int row_dilation, int col_dilation, int row_stride,
Dconv_ops.cc72 int row_dilation, int col_dilation, const Padding& padding, in operator ()()
99 col_dilation == 1 && padding == VALID) { in operator ()()
116 row_dilation, col_dilation, BrainPadding2EigenPadding(padding)); in operator ()()
126 int col_dilation, int row_stride, int col_stride, in operator ()()
146 row_dilation, col_dilation, padding, output, in operator ()()
581 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
611 row_dilation == 1 && col_dilation == 1 && row_stride == 1 && in operator ()()
639 col_dilation == 1 && padding == VALID && in operator ()()
687 status = GetWindowedOutputSizeVerboseV2(in_cols, patch_cols, col_dilation, in operator ()()
782 .set_horizontal_dilation_rate(col_dilation) 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_filter_ops.cc105 int row_dilation, int col_dilation, int row_stride, in operator ()()
546 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
555 dilations[GetTensorDimIndex(data_format, 'W')] = col_dilation; in operator ()()
587 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
Dconv_grad_input_ops.cc111 int row_dilation, int col_dilation, int row_stride, in operator ()()
683 int col_dilation, int row_stride, int col_stride, const Padding& padding, in operator ()() argument
697 dilations[input_w] = col_dilation; in operator ()()
726 col_dilation, col_stride, padding, &expected_out_cols, &padding_left, in operator ()()
Dconv_ops_fused_impl.h249 int row_dilation, int col_dilation, // in LaunchFusedConv2DWithOutputKernel() argument
254 col_dilation_(col_dilation), in LaunchFusedConv2DWithOutputKernel()