Searched refs:kActivationParam (Results 1 – 15 of 15) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | SVDF.h | 51 static constexpr int kActivationParam = 6; variable
|
| D | RNN.h | 42 static constexpr int kActivationParam = 5; variable
|
| D | UnidirectionalSequenceRNN.cpp | 32 constexpr uint32_t kActivationParam = 5; variable
66 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
|
| D | BidirectionalSequenceRNN.cpp | 43 constexpr uint32_t kActivationParam = 12; variable
109 int32_t activation = context->getInputValue<int32_t>(kActivationParam); in executeTyped()
284 kBwHiddenStateTensor, kActivationParam, kTimeMajorParam, kMergeOutputsParam, in prepare()
|
| D | UnidirectionalSequenceLSTM.cpp | 68 constexpr uint32_t kActivationParam = 20; variable
98 static_cast<TfLiteFusedActivation>(context->getInputValue<int32_t>(kActivationParam)); in getLSTMParams()
174 kActivationParam, in prepare()
|
| D | BidirectionalSequenceLSTM.h | 130 static constexpr int kActivationParam = 48; variable
|
| D | LSTM.h | 89 static constexpr int kActivationParam = 20; variable
|
| D | RNNTest.cpp | 241 ASSERT_EQ(execution.setInput(RNN::kActivationParam, &activation_, in Invoke()
|
| D | RNN.cpp | 38 getScalarData<int32_t>(operands[operation.inputs[kActivationParam]])); in RNN()
|
| D | SVDF.cpp | 39 *GetInput(operation, operands, kActivationParam))); in SVDF()
|
| D | SVDFTest.cpp | 286 ASSERT_EQ(execution.setInput(SVDF::kActivationParam, &activation, in Invoke()
|
| D | LayerNormLSTMTest.cpp | 252 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, &activation_, sizeof(activation_)), in Invoke()
|
| D | LSTMTest.cpp | 236 ASSERT_EQ(execution.setInput(LSTMCell::kActivationParam, in Invoke()
|
| D | BidirectionalSequenceLSTM.cpp | 140 getScalarData<int32_t>(*GetInput(operation, operands, kActivationParam))); in BidirectionalSequenceLSTM()
|
| D | LSTM.cpp | 82 getScalarData<int32_t>(*GetInput(operation, operands, kActivationParam))); in LSTMCell()
|