Lines Matching defs:output_cols
190 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1);
198 c->set_output(0, c->Matrix(output_rows, output_cols));
464 DimensionHandle output_rows, output_cols;
470 padding, &output_cols));
474 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols},
531 DimensionHandle output_planes, output_rows, output_cols;
539 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols));
544 output_planes, output_rows, output_cols});
547 output_cols, output_depth_dim});
608 DimensionHandle output_rows, output_cols;
613 c, in_cols_dim, filter_cols_dim, stride_cols, padding, &output_cols));
618 c->MakeShape({batch_size_dim, output_depth, output_rows, output_cols});
621 c->MakeShape({batch_size_dim, output_rows, output_cols, output_depth});
682 DimensionHandle output_rows, output_cols;
686 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols));
690 {output_rows, output_cols}, depth_dim,
831 DimensionHandle output_rows, output_cols, output_depth;
835 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols));
840 {output_rows, output_cols},
933 DimensionHandle output_rows, output_cols, output_depth;
937 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols));
942 {output_rows, output_cols},
1010 DimensionHandle output_planes, output_rows, output_cols;
1016 c, in_cols_dim, kernel_cols, stride_cols, padding, &output_cols));
1021 output_planes, output_rows, output_cols});
1024 output_cols, output_depth_dim});
