Home
last modified time | relevance | path

Searched refs:kOutputStateOutTensor (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/common/cpu_operations/
DUnidirectionalSequenceLSTM.cpp272 NN_RET_CHECK(!context->isOmittedOutput(kOutputStateOutTensor)); in prepare()
279 NN_RET_CHECK(context->setOutputShape(kOutputStateOutTensor, outputStateOutTensor)); in prepare()
307 outputStateOut = context->getOutputBuffer<float>(kOutputStateOutTensor); in execute()
359 outputStateOut = context->getOutputBuffer<_Float16>(kOutputStateOutTensor); in execute()
DQLSTM.cpp245 return context->setOutputShape(kOutputStateOutTensor, outputShape) && in prepare()
347 reinterpret_cast<uint8_t*>(context->getOutputBuffer(kOutputStateOutTensor)); in execute()
DLSTM.cpp124 output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor); in LSTMCell()
/packages/modules/NeuralNetworks/common/types/operations/include/
DUnidirectionalSequenceLSTM.h79 constexpr uint32_t kOutputStateOutTensor = 1; variable
DQLSTM.h87 constexpr uint32_t kOutputStateOutTensor = 0; variable
DLSTM.h102 static constexpr int kOutputStateOutTensor = 1; variable
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1043 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()