Searched refs:InferTriangularSolveShape (Results 1 – 5 of 5) sorted by relevance
120 static StatusOr<Shape> InferTriangularSolveShape(
231 ShapeInference::InferTriangularSolveShape( in HandleTriangularSolve()
1963 /* static */ StatusOr<Shape> ShapeInference::InferTriangularSolveShape( in InferTriangularSolveShape() function in xla::ShapeInference
2061 return ShapeInference::InferTriangularSolveShape( in CreateInstruction()
4389 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferTriangularSolveShape( in TriangularSolve()