Searched refs:kFwOutputTensor (Results 1 – 4 of 4) sorted by relevance
47 constexpr uint32_t kFwOutputTensor = 0; variable113 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped()114 Shape fwOutputShape = context->getOutputShape(kFwOutputTensor); in executeTyped()238 context->getOutputBuffer<T>(kFwOutputTensor)); in executeTyped()360 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare()365 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
148 static constexpr int kFwOutputTensor = 0; variable
154 fw_output_ = GetOutput(operation, operands, kFwOutputTensor); in BidirectionalSequenceLSTM()
1104 mOperands[outs[BidirectionalSequenceLSTM::kFwOutputTensor]]; in executeOperation()