Searched refs:ActivationFunctor (Results 1 – 2 of 2) sorted by relevance
35 class ActivationFunctor {37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor() function
216 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()