Searched refs:TriangularSolveExpander (Results 1 – 5 of 5) sorted by relevance
238 XlaOp TriangularSolveExpander::InvertDiagonalBlocks( in InvertDiagonalBlocks()365 XlaOp TriangularSolveExpander::SolveByInvertingDiagonalBlocks( in SolveByInvertingDiagonalBlocks()409 XlaOp TriangularSolveExpander::SolveDirectly( in SolveDirectly()466 XlaOp TriangularSolveExpander::BuildTriangularSolve( in BuildTriangularSolve()545 TriangularSolveExpander::TriangularSolveExpander(int64_t block_size) in TriangularSolveExpander() function in xla::TriangularSolveExpander550 bool TriangularSolveExpander::InstructionMatchesPattern( in InstructionMatchesPattern()555 StatusOr<HloInstruction*> TriangularSolveExpander::ExpandInstruction( in ExpandInstruction()
25 class TriangularSolveExpander : public OpExpanderPass {27 explicit TriangularSolveExpander(int64_t block_size = 128);
52 TriangularSolveExpander triangular_solve_expander(block_size); in TEST_P()
90 pipeline.AddPass<TriangularSolveExpander>(); in RunHloOptimization()
481 pipeline.AddPass<TriangularSolveExpander>(); in RunHloPassesThroughLayoutAssn()