Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/common/operations/
DUnidirectionalSequenceLSTM.cpp88 constexpr uint32_t kOutputStateOutTensor = 1; variable
381 NN_RET_CHECK(!context->isOmittedOutput(kOutputStateOutTensor)); in prepare()
388 NN_RET_CHECK(context->setOutputShape(kOutputStateOutTensor, outputStateOutTensor)); in prepare()
416 outputStateOut = context->getOutputBuffer<float>(kOutputStateOutTensor); in execute()
468 outputStateOut = context->getOutputBuffer<_Float16>(kOutputStateOutTensor); in execute()
DQLSTM.cpp97 constexpr uint32_t kOutputStateOutTensor = 0; variable
361 return context->setOutputShape(kOutputStateOutTensor, outputShape) && in prepare()
463 reinterpret_cast<uint8_t*>(context->getOutputBuffer(kOutputStateOutTensor)); in execute()
DLSTM.h103 static constexpr int kOutputStateOutTensor = 1; variable
DLSTM.cpp124 output_state_out_ = GetOutput(operation, operands, kOutputStateOutTensor); in LSTMCell()
/packages/modules/NeuralNetworks/common/
DCpuExecutor.cpp1042 RunTimeOperandInfo& outputStateOut = operands[outs[LSTMCell::kOutputStateOutTensor]]; in executeOperation()