Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/types/include/
DActivationFunctor.h35 class ActivationFunctor {
37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor() function
/packages/modules/NeuralNetworks/common/
DAndroid.bp132 "ActivationFunctor.cpp",
334 "ActivationFunctor.cpp",
/packages/modules/NeuralNetworks/common/cpu_operations/
DRNN.cpp212 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()