Searched refs:kCellStateOutTensor (Results 1 – 10 of 10) sorted by relevance
273 NN_RET_CHECK(!context->isOmittedOutput(kCellStateOutTensor)); in prepare()285 NN_RET_CHECK(context->setOutputShape(kCellStateOutTensor, cellStateOutTensor)); in prepare()308 cellStateOut = context->getOutputBuffer<float>(kCellStateOutTensor); in execute()360 cellStateOut = context->getOutputBuffer<_Float16>(kCellStateOutTensor); in execute()
242 Shape cellStateOutShape = context->getOutputShape(kCellStateOutTensor); in prepare()246 context->setOutputShape(kCellStateOutTensor, cellStateOutShape) && in prepare()349 reinterpret_cast<int16_t*>(context->getOutputBuffer(kCellStateOutTensor)); in execute()
141 setOutputTensor(&execution, QuantizedLSTMCell::kCellStateOutTensor, &cellStateOut_), in invoke()
245 cellStateOut_ = GetOutput(operation, operands, kCellStateOutTensor); in QuantizedLSTMCell()
125 cell_state_out_ = GetOutput(operation, operands, kCellStateOutTensor); in LSTMCell()
61 static constexpr int kCellStateOutTensor = 0; variable
80 constexpr uint32_t kCellStateOutTensor = 2; variable
88 constexpr uint32_t kCellStateOutTensor = 1; variable
103 static constexpr int kCellStateOutTensor = 2; variable
1044 RunTimeOperandInfo& cellStateOut = operands[outs[LSTMCell::kCellStateOutTensor]]; in executeOperation()1621 operands[outs[QuantizedLSTMCell::kCellStateOutTensor]]; in executeOperation()