Searched refs:LogisticExpansionType (Results 1 – 5 of 5) sorted by relevance
27 enum class LogisticExpansionType { enum35 explicit LogisticExpander(LogisticExpansionType expansion_type) in LogisticExpander()48 LogisticExpansionType expansion_type_;
81 case LogisticExpansionType::kTanh: in ExpandInstruction()83 case LogisticExpansionType::kExp: in ExpandInstruction()
65 LogisticExpander logistic_expander(LogisticExpansionType::kTanh); in TEST_F()92 LogisticExpander logistic_expander(LogisticExpansionType::kExp); in TEST_F()
338 /*expansion_type=*/LogisticExpansionType::kExp); in RunHloPassesThroughLayoutAssn()
214 /*expansion_type=*/LogisticExpansionType::kExp); in OptimizeHloModule()