Searched refs:ScatterExpander (Results 1 – 6 of 6) sorted by relevance
64 ScatterExpander scatter_expander(ScatterExpander::kEliminateAllScatters); in TEST_F()99 ScatterExpander scatter_expander(ScatterExpander::kEliminateSimpleScatters); in TEST_F()132 ScatterExpander scatter_expander(ScatterExpander::kEliminateSimpleScatters); in TEST_F()
38 class ScatterExpander : public OpExpanderPass {45 explicit ScatterExpander(Mode m) : mode_(m) {} in ScatterExpander() function
361 StatusOr<HloInstruction*> ScatterExpander::ExpandInstruction( in ExpandInstruction()421 bool ScatterExpander::InstructionMatchesPattern(HloInstruction* inst) { in InstructionMatchesPattern()
24 class GpuScatterExpander : public ScatterExpander {28 GpuScatterExpander() : ScatterExpander(kEliminateAllScatters) {} in GpuScatterExpander()
230 pass.AddPass<ScatterExpander>(ScatterExpander::kEliminateSimpleScatters); in OptimizeHloModule()
341 pipeline.AddPass<ScatterExpander>(ScatterExpander::kEliminateAllScatters); in RunHloPassesThroughLayoutAssn()