Searched refs:MatrixDiagV2Shape (Results 1 – 3 of 3) sorted by relevance
169 Status MatrixDiagV2Shape(shape_inference::InferenceContext* c);
1321 Status MatrixDiagV2Shape(shape_inference::InferenceContext* c) { in MatrixDiagV2Shape() function
841 .SetShapeFn(shape_inference::MatrixDiagV2Shape);854 .SetShapeFn(shape_inference::MatrixDiagV2Shape);