Searched defs:diag_index_shape (Results 1 – 2 of 2) sorted by relevance
59 TensorShape diag_index_shape = context->InputShape("k"); in ProcessDiagIndex() local
1360 ShapeHandle input_shape, diag_index_shape, unused_shape; in MatrixDiagPartV2Shape() local1414 ShapeHandle input_shape, diag_index_shape, unused_shape; in MatrixDiagV2Shape() local1514 ShapeHandle input_shape, diag_shape, diag_index_shape; in MatrixSetDiagV2Shape() local