Searched refs:fwOutput (Results 1 – 3 of 3) sorted by relevance
133 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped() local171 fwOutput = fwOutputTransposed.data(); in executeTyped()234 T* fwOutputBatchPtr = fwOutput + i * batchSize * fwOutputBatchStride; in executeTyped()258 bwOutputBatchPtr = fwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()379 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare() local380 fwOutput.dimensions.resize(3); in prepare()381 fwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()382 fwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()383 fwOutput.dimensions[2] = mergeOutputs ? fwNumUnits + bwNumUnits : fwNumUnits; in prepare()384 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
1002 RunTimeOperandInfo& fwOutput = in executeOperation() local1011 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()
3709 * * 0: fwOutput.