Searched refs:GatherExpander (Results 1 – 5 of 5) sorted by relevance
46 Status status = GatherExpander{GatherExpander::kEliminateAllGathers} in TEST_F()76 GatherExpander{GatherExpander::kEliminateAllGathers}.Run(module.get())); in TEST_F()139 GatherExpander{GatherExpander::kEliminateAllGathers}.Run(module.get())); in TEST_F()176 GatherExpander pass(GatherExpander::kEliminateSimpleGathers); in TEST_F()199 GatherExpander pass(GatherExpander::kEliminateAllGathers); in TEST_F()
39 class GatherExpander : public OpExpanderPass {46 explicit GatherExpander(Mode m) : mode_(m) {} in GatherExpander() function
323 StatusOr<HloInstruction*> GatherExpander::ExpandInstruction( in ExpandInstruction()381 bool GatherExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern()
355 pass.AddPass<GatherExpander>(GatherExpander::kEliminateSimpleGathers); in RunHloPassesThroughLayoutAssn()
229 pass.AddPass<GatherExpander>(GatherExpander::kEliminateSimpleGathers); in OptimizeHloModule()