Searched refs:MatrixDiagPartV2Shape (Results 1 – 3 of 3) sorted by relevance
166 Status MatrixDiagPartV2Shape(shape_inference::InferenceContext* c);
1268 Status MatrixDiagPartV2Shape(shape_inference::InferenceContext* c) { in MatrixDiagPartV2Shape() function
939 .SetShapeFn(shape_inference::MatrixDiagPartV2Shape);950 .SetShapeFn(shape_inference::MatrixDiagPartV2Shape);