Searched refs:activation_ (Results 1 – 7 of 7) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | RNNTest.cpp | 134 : batches_(batches), units_(units), input_size_(size), activation_(kActivationRelu) { in BasicRNNOpModel()
215 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
228 const int activation_; member in android::nn::wrapper::BasicRNNOpModel
|
| D | LayerNormLSTMTest.cpp | 96 activation_(ActivationFn::kActivationTanh), in LayerNormLSTMOpModel()
251 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
272 const int activation_; member in android::nn::wrapper::LayerNormLSTMOpModel
|
| D | RNN.cpp | 42 activation_ = static_cast<ActivationFn>( in RNN()
93 recurrent_weights_->shape(), activation_, in Eval()
105 recurrent_weights_->shape(), activation_, in Eval()
|
| D | LSTMTest.cpp | 89 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()
240 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
261 const int activation_; member in android::nn::wrapper::LSTMOpModel
|
| D | SVDF.h | 33 TfLiteFusedActivation activation_; member
|
| D | RNN.h | 65 ActivationFn activation_;
|
| D | SVDF.cpp | 44 params_.activation_ = static_cast<TfLiteFusedActivation>(getScalarDataWithDefault<int>( in SVDF()
224 params_.activation_, output_ptr_batch); in EvalFloat32()
|