Home
last modified time | relevance | path

Searched refs:ActivationFn (Results 1 – 6 of 6) sorted by relevance

/frameworks/ml/nn/common/include/
DActivationFunctor.h25 enum ActivationFn { enum
37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor()
59 ActivationFn act_;
/frameworks/ml/nn/common/operations/
DRNN.h65 ActivationFn activation_;
DRNN.cpp42 activation_ = static_cast<ActivationFn>( in RNN()
216 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()
DActivation.cpp123 template <ActivationFn activation>
218 template <ActivationFn activation>
DLayerNormLSTMTest.cpp96 activation_(ActivationFn::kActivationTanh), in LayerNormLSTMOpModel()
DLSTMTest.cpp89 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()