| common_shape_fns.cc | 429 DimensionHandle batch_size_dim; local
433 &batch_size_dim, &input_spatial_dims,
474 ShapeFromDimensions(batch_size_dim, {output_rows, output_cols},
515 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); local
543 output_shape = c->MakeShape({batch_size_dim, output_depth_dim,
546 output_shape = c->MakeShape({batch_size_dim, output_planes, output_rows,
586 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); local
618 c->MakeShape({batch_size_dim, output_depth, output_rows, output_cols});
621 c->MakeShape({batch_size_dim, output_rows, output_cols, output_depth});
666 DimensionHandle batch_size_dim = c->Dim local
818 DimensionHandle batch_size_dim = c->Dim( local
920 DimensionHandle batch_size_dim = c->Dim( local
998 DimensionHandle batch_size_dim = c->Dim(input_shape, 0); local
[all...] |
