Searched refs:InferTriangularSolveShape (Results 1 – 4 of 4) sorted by relevance
120 static StatusOr<Shape> InferTriangularSolveShape(
174 ShapeInference::InferTriangularSolveShape( in HandleTriangularSolve()
1903 /* static */ StatusOr<Shape> ShapeInference::InferTriangularSolveShape( in InferTriangularSolveShape() function in xla::ShapeInference
3043 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferTriangularSolveShape( in TriangularSolve()