Searched refs:kBwOutputTensor (Results 1 – 5 of 5) sorted by relevance
138 bwOutputShape = context->getOutputShape(kBwOutputTensor); in executeTyped()139 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()280 context->getOutputBuffer<T>(kBwOutputTensor)); in executeTyped()386 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare()391 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
193 bw_output_ = GetOutput(operation, operands, kBwOutputTensor); in BidirectionalSequenceLSTM()
51 constexpr uint32_t kBwOutputTensor = 1; // Only if mergeOutputs parameter is false variable
148 static constexpr int kBwOutputTensor = 1; // Ignored if merge_outputs is set. variable
1014 operands[outs[BidirectionalSequenceLSTM::kBwOutputTensor]]; in executeOperation()