Searched refs:activation_ (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNNTest.cpp | 156 activation_(kActivationRelu) { in BasicRNNOpModel()
241 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, in Invoke()
242 sizeof(activation_)), in Invoke()
255 const int activation_; member in android::nn::wrapper::BasicRNNOpModel
|
| D | RNN.cpp | 37 activation_ = static_cast<ActivationFn>( in RNN()
93 recurrent_weights_->shape(), activation_, in Eval()
105 recurrent_weights_->shape(), activation_, in Eval()
|
| D | LayerNormLSTMTest.cpp | 97 activation_(ActivationFn::kActivationTanh), in LayerNormLSTMOpModel()
252 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
273 const int activation_; member in android::nn::wrapper::LayerNormLSTMOpModel
|
| D | LSTMTest.cpp | 86 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()
237 &activation_, sizeof(activation_)), in Invoke()
260 const int activation_; member in android::nn::wrapper::LSTMOpModel
|
| D | SVDF.h | 31 TfLiteFusedActivation activation_; member
|
| D | RNN.h | 63 ActivationFn activation_;
|
| D | SVDF.cpp | 38 params_.activation_ = static_cast<TfLiteFusedActivation>(getScalarData<int>( in SVDF()
208 params_.activation_, output_ptr_batch); in EvalFloat32()
|