Searched refs:TriangularSolveExpander (Results 1 – 5 of 5) sorted by relevance
25 class TriangularSolveExpander : public OpExpanderPass {27 explicit TriangularSolveExpander(int64_t block_size = 128);
244 XlaOp TriangularSolveExpander::InvertDiagonalBlocks( in InvertDiagonalBlocks()372 XlaOp TriangularSolveExpander::SolveByInvertingDiagonalBlocks( in SolveByInvertingDiagonalBlocks()416 XlaOp TriangularSolveExpander::SolveDirectly( in SolveDirectly()473 XlaOp TriangularSolveExpander::BuildTriangularSolve( in BuildTriangularSolve()552 TriangularSolveExpander::TriangularSolveExpander(int64_t block_size) in TriangularSolveExpander() function in xla::TriangularSolveExpander557 bool TriangularSolveExpander::InstructionMatchesPattern( in InstructionMatchesPattern()562 StatusOr<HloInstruction*> TriangularSolveExpander::ExpandInstruction( in ExpandInstruction()
52 TriangularSolveExpander triangular_solve_expander(block_size); in TEST_P()
90 pipeline.AddPass<TriangularSolveExpander>(); in RunHloOptimization()
329 pipeline.AddPass<TriangularSolveExpander>(); in RunHloPassesThroughLayoutAssn()