Searched refs:kFwOutputTensor (Results 1 – 4 of 4) sorted by relevance
57 constexpr uint32_t kFwOutputTensor = 0; variable164 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped()165 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped()308 context->getOutputBuffer<T>(kFwOutputTensor)); in executeTyped()439 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare()444 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
150 static constexpr int kFwOutputTensor = 0; variable
192 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
1058 operands[outs[BidirectionalSequenceLSTM::kFwOutputTensor]]; in executeOperation()