Searched refs:fwOutput (Results 1 – 5 of 5) sorted by relevance
162 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped() local200 fwOutput = fwOutputTransposed.data(); in executeTyped()263 T* fwOutputBatchPtr = fwOutput + i * batchSize * fwOutputBatchStride; in executeTyped()287 bwOutputBatchPtr = fwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()438 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare() local439 fwOutput.dimensions.resize(3); in prepare()440 fwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()441 fwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()442 fwOutput.dimensions[2] = mergeOutputs ? fwNumUnits + bwNumUnits : fwNumUnits; in prepare()443 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
1001 RunTimeOperandInfo& fwOutput = in executeOperation() local1010 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()
3652 * * 0: fwOutput.
2652 * * 0: fwOutput.
2801 * * 0: fwOutput.