Searched refs:activationOperand (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | SVDF.cpp | 43 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in SVDF() local
45 activationOperand, TfLiteFusedActivation::kTfLiteActNone)); in SVDF()
71 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare() local
72 NN_RET_CHECK(activationOperand.length >= sizeof(int)); in Prepare()
|
| D | BidirectionalSequenceLSTM.cpp | 172 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in BidirectionalSequenceLSTM() local
174 activationOperand, TfLiteFusedActivation::kTfLiteActNone)); in BidirectionalSequenceLSTM()
250 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare() local
251 NN_RET_CHECK(activationOperand.length >= sizeof(int32_t)); in Prepare()
|
| D | LSTM.cpp | 86 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in LSTMCell() local
88 activationOperand, TfLiteFusedActivation::kTfLiteActNone)); in LSTMCell()
333 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare() local
334 NN_RET_CHECK(activationOperand.length >= sizeof(int32_t)); in Prepare()
|