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