Searched refs:InferTriangularSolveShape (Results 1 – 5 of 5) sorted by relevance
122 static StatusOr<Shape> InferTriangularSolveShape(
171 ShapeInference::InferTriangularSolveShape( in HandleTriangularSolve()
1962 /* static */ StatusOr<Shape> ShapeInference::InferTriangularSolveShape( in InferTriangularSolveShape() function in xla::ShapeInference
1689 return ShapeInference::InferTriangularSolveShape( in ParseInstructionRhs()
4433 TF_ASSIGN_OR_RETURN(Shape shape, ShapeInference::InferTriangularSolveShape( in TriangularSolve()