Searched refs:activationOperand (Results 1 – 6 of 6) sorted by relevance
/packages/modules/NeuralNetworks/runtime/operation_converters/ |
D | AddOperationConverter.cpp | 40 const Operand& activationOperand = in convert() local 42 NN_RET_CHECK(isOperandConstant(activationOperand)); in convert() 44 context->getConstantScalar<int32_t>(activationOperand)); in convert()
|
D | DepthwiseConv2DOperationConverter.cpp | 68 const Operand& activationOperand = in convert() local 74 NN_RET_CHECK(isOperandConstant(activationOperand)); in convert() 82 context->getConstantScalar<int32_t>(activationOperand)); in convert()
|
D | Conv2DOperationConverter.cpp | 186 const Operand& activationOperand = in convert() local 190 NN_RET_CHECK(isOperandConstant(activationOperand)); in convert() 196 context->getConstantScalar<int32_t>(activationOperand)); in convert()
|
/packages/modules/NeuralNetworks/common/cpu_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 | 171 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in BidirectionalSequenceLSTM() local 173 activationOperand, TfLiteFusedActivation::kTfLiteActNone)); in BidirectionalSequenceLSTM() 249 const auto& activationOperand = *GetInput(operation, operands, kActivationParam); in Prepare() local 250 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()
|