Searched refs:fwOutput (Results 1 – 3 of 3) sorted by relevance
164 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped() local202 fwOutput = fwOutputTransposed.data(); in executeTyped()265 T* fwOutputBatchPtr = fwOutput + i * batchSize * fwOutputBatchStride; in executeTyped()289 bwOutputBatchPtr = fwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()439 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare() local440 fwOutput.dimensions.resize(3); in prepare()441 fwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()442 fwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()443 fwOutput.dimensions[2] = mergeOutputs ? fwNumUnits + bwNumUnits : fwNumUnits; in prepare()444 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
1057 RunTimeOperandInfo& fwOutput = in executeOperation() local1066 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()
3546 * * 0: fwOutput.