Searched refs:ActivationFn (Results 1 – 6 of 6) sorted by relevance
25 enum ActivationFn { enum37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor()59 ActivationFn act_;
38 activation_ = static_cast<ActivationFn>( in RNN()212 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()
65 ActivationFn activation_;
124 template <ActivationFn activation>219 template <ActivationFn activation>
96 activation_(ActivationFn::kActivationTanh), in LayerNormLSTMOpModel()
88 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()