Searched refs:ActivationFn (Results 1 – 4 of 4) sorted by relevance
25 enum ActivationFn { enum37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor()60 ActivationFn act_;
60 ActivationFn activation_;
33 activation_ = static_cast<ActivationFn>( in RNN()
86 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()