Searched refs:ActivationFn (Results 1 – 14 of 14) sorted by relevance
21 static_assert(static_cast<int32_t>(ActivationFn::kActivationNone) == kTfLiteActNone);22 static_assert(static_cast<int32_t>(ActivationFn::kActivationRelu) == kTfLiteActRelu);23 static_assert(static_cast<int32_t>(ActivationFn::kActivationRelu1) == kTfLiteActReluN1To1);24 static_assert(static_cast<int32_t>(ActivationFn::kActivationRelu6) == kTfLiteActRelu6);25 static_assert(static_cast<int32_t>(ActivationFn::kActivationTanh) == kTfLiteActTanh);26 static_assert(static_cast<int32_t>(ActivationFn::kActivationSignBit) == kTfLiteActSignBit);27 static_assert(static_cast<int32_t>(ActivationFn::kActivationSigmoid) == kTfLiteActSigmoid);
25 enum ActivationFn { enum37 explicit ActivationFunctor(ActivationFn act) : act_(act) {} in ActivationFunctor()59 ActivationFn act_;
33 ActivationFn activation_;
71 ActivationFn activation_;
32 ActivationFn activation;
38 activation_ = static_cast<ActivationFn>( in RNN()212 (ActivationFunctor(static_cast<ActivationFn>(activation)))(output_ptr_batch[o]); in RNNStep()
44 params_.activation_ = static_cast<ActivationFn>(getScalarDataWithDefault<int>( in SVDF()
126 template <ActivationFn activation>221 template <ActivationFn activation>
266 int activation = ActivationFn::kActivationNone; in Invoke()
97 activation_(ActivationFn::kActivationTanh), in LayerNormLSTMOpModel()
88 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()
52 static_cast<ActivationFn>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
172 params_.activation = static_cast<ActivationFn>(getScalarDataWithDefault<int32_t>( in BidirectionalSequenceLSTM()
87 params_.activation = static_cast<ActivationFn>(getScalarDataWithDefault<int32_t>( in LSTMCell()