Searched refs:kActivationParam (Results 1 – 15 of 15) sorted by relevance
| /frameworks/ml/nn/common/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 | SVDF.h | 53 static constexpr int kActivationParam = 6; variable
|
| D | RNN.h | 44 static constexpr int kActivationParam = 5; variable
|
| D | UnidirectionalSequenceRNN.cpp | 37 constexpr uint32_t kActivationParam = 5; variable
75 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
|
| D | BidirectionalSequenceRNN.cpp | 48 constexpr uint32_t kActivationParam = 12; variable
160 const int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
361 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 70 constexpr uint32_t kActivationParam = 20; variable
105 static_cast<TfLiteFusedActivation>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
187 kActivationParam, in prepare()
|
| D | BidirectionalSequenceLSTM.h | 132 static constexpr int kActivationParam = 48; variable
|
| D | LSTM.h | 91 static constexpr int kActivationParam = 20; variable
|
| D | BidirectionalSequenceLSTM.cpp | 172 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in BidirectionalSequenceLSTM()
238 kActivationParam, in Prepare()
250 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare()
|
| D | RNNTest.cpp | 215 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| D | SVDFTest.cpp | 265 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation, sizeof(activation)), in Invoke()
|
| D | RNN.cpp | 43 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 | 251 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| D | LSTMTest.cpp | 240 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|