Home
last modified time | relevance | path

Searched refs:ActivationFunctor (Results 1 – 2 of 2) sorted by relevance

/packages/modules/NeuralNetworks/common/include/
DActivationFunctor.h35 class ActivationFunctor {
37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor() function
/packages/modules/NeuralNetworks/common/operations/
DRNN.cpp212 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()