Searched refs:activation_ (Results 1 – 7 of 7) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | RNNTest.cpp | 135 : batches_(batches), units_(units), input_size_(size), activation_(kActivationRelu) { in BasicRNNOpModel()
216 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
229 const int activation_; member in android::nn::wrapper::BasicRNNOpModel
|
| 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 | RNN.cpp | 38 activation_ = static_cast<ActivationFn>( in RNN()
89 recurrent_weights_->shape(), activation_, in Eval()
101 recurrent_weights_->shape(), activation_, in Eval()
|
| D | LSTMTest.cpp | 88 activation_(ActivationFn::kActivationTanh), in LSTMOpModel()
239 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
260 const int activation_; member in android::nn::wrapper::LSTMOpModel
|
| D | SVDF.cpp | 44 params_.activation_ = static_cast<ActivationFn>(getScalarDataWithDefault<int>( in SVDF()
218 static_cast<TfLiteFusedActivation>(params_.activation_), outputData); in EvalFloat32()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | SVDF.h | 33 ActivationFn activation_; member
|
| D | RNN.h | 71 ActivationFn activation_;
|