Searched refs:output_col_dim (Results 1 – 1 of 1) sorted by relevance
1406 const DimensionHandle output_col_dim = c->MakeDim(num_cols); in MatrixDiagV2Shape() local1411 c->Concatenate(output_shape, c->Vector(output_col_dim), &output_shape)); in MatrixDiagV2Shape()1416 output_col_dim, &output_shape)); in MatrixDiagV2Shape()