Searched refs:LogisticExpander (Results 1 – 5 of 5) sorted by relevance
33 class LogisticExpander : public OpExpanderPass {35 explicit LogisticExpander(LogisticExpansionType expansion_type) in LogisticExpander() function37 ~LogisticExpander() override = default;
74 bool LogisticExpander::InstructionMatchesPattern(HloInstruction* instruction) { in InstructionMatchesPattern()78 StatusOr<HloInstruction*> LogisticExpander::ExpandInstruction( in ExpandInstruction()
65 LogisticExpander logistic_expander(LogisticExpansionType::kTanh); in TEST_F()92 LogisticExpander logistic_expander(LogisticExpansionType::kExp); in TEST_F()
337 pipeline.AddPass<LogisticExpander>( in RunHloPassesThroughLayoutAssn()
213 pipeline.AddPass<LogisticExpander>( in OptimizeHloModule()