Searched refs:kCellStateOutTensor (Results 1 – 8 of 8) sorted by relevance
87 constexpr uint32_t kCellStateOutTensor = 2; variable379 NN_RET_CHECK(!context->isOmittedOutput(kCellStateOutTensor)); in prepare()391 NN_RET_CHECK(context->setOutputShape(kCellStateOutTensor, cellStateOutTensor)); in prepare()414 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()466 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
61 static constexpr int kCellStateOutTensor = 0; variable
95 constexpr uint32_t kCellStateOutTensor = 1; variable357 Shape cellStateOutShape = context->getOutputShape(kCellStateOutTensor); in prepare()361 context->setOutputShape(kCellStateOutTensor, cellStateOutShape) && in prepare()463 reinterpret_cast<int16_t*>(context->getOutputBuffer(kCellStateOutTensor)); in execute()
104 static constexpr int kCellStateOutTensor = 2; variable
140 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()
244 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()
125 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
1097 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()1675 operands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()