| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | SVDF.cpp | 43 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in SVDF()
60 kRankParam, kActivationParam, in Prepare()
71 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
|
| D | BidirectionalSequenceLSTM.cpp | 171 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in BidirectionalSequenceLSTM()
237 kActivationParam, in Prepare()
249 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 52 static_cast<ActivationFn>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
80 kActivationParam, in prepare()
|
| D | BidirectionalSequenceRNN.cpp | 129 const int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
301 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
|
| D | RNNTest.cpp | 216 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| D | UnidirectionalSequenceRNN.cpp | 62 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
|
| D | SVDFTest.cpp | 267 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation, sizeof(activation)), in Invoke()
|
| D | RNN.cpp | 39 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]])); in RNN()
|
| D | LSTM.cpp | 86 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in LSTMCell()
322 kActivationParam, in Prepare()
333 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
|
| D | LayerNormLSTMTest.cpp | 252 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| D | LSTMTest.cpp | 239 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| /packages/modules/NeuralNetworks/common/types/operations/include/ |
| D | UnidirectionalSequenceRNN.h | 30 constexpr uint32_t kActivationParam = 5; variable
|
| D | SVDF.h | 53 static constexpr int kActivationParam = 6; variable
|
| D | BidirectionalSequenceRNN.h | 41 constexpr uint32_t kActivationParam = 12; variable
|
| D | UnidirectionalSequenceLSTM.h | 63 constexpr uint32_t kActivationParam = 20; variable
|
| D | RNN.h | 50 static constexpr int kActivationParam = 5; variable
|
| D | BidirectionalSequenceLSTM.h | 129 static constexpr int kActivationParam = 48; variable
|
| D | LSTM.h | 90 static constexpr int kActivationParam = 20; variable
|