Searched refs:MatrixSetDiagV2Shape (Results 1 – 3 of 3) sorted by relevance
172 Status MatrixSetDiagV2Shape(shape_inference::InferenceContext* c);
1422 Status MatrixSetDiagV2Shape(shape_inference::InferenceContext* c) { in MatrixSetDiagV2Shape() function
895 .SetShapeFn(shape_inference::MatrixSetDiagV2Shape);906 .SetShapeFn(shape_inference::MatrixSetDiagV2Shape);