Searched refs:TriangularSolveExpander (Results 1 – 5 of 5) sorted by relevance
25 class TriangularSolveExpander : public OpExpanderPass {27 explicit TriangularSolveExpander(int64 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()551 TriangularSolveExpander::TriangularSolveExpander(int64 block_size) in TriangularSolveExpander() function in xla::TriangularSolveExpander556 bool TriangularSolveExpander::InstructionMatchesPattern( in InstructionMatchesPattern()561 StatusOr<HloInstruction*> TriangularSolveExpander::ExpandInstruction( in ExpandInstruction()
52 TriangularSolveExpander triangular_solve_expander(block_size); in TEST_P()
88 pipeline.AddPass<TriangularSolveExpander>(); in RunHloOptimization()
308 pipeline.AddPass<TriangularSolveExpander>(); in RunHloPassesThroughLayoutAssn()