Searched refs:fwOutput (Results 1 – 2 of 2) sorted by relevance
113 T* fwOutput = context->getOutputBuffer<T>(kFwOutputTensor); in executeTyped() local151 fwOutput = fwOutputTransposed.data(); in executeTyped()194 T* fwOutputBatchPtr = fwOutput + i * batchSize * fwOutputBatchStride; in executeTyped()219 bwOutputBatchPtr = fwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()360 Shape fwOutput = context->getOutputShape(kFwOutputTensor); in prepare() local361 fwOutput.dimensions.resize(3); in prepare()362 fwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()363 fwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()364 fwOutput.dimensions[2] = mergeOutputs ? fwNumUnits + bwNumUnits : fwNumUnits; in prepare()365 NN_RET_CHECK(context->setOutputShape(kFwOutputTensor, fwOutput)); in prepare()
1103 RunTimeOperandInfo& fwOutput = in executeOperation() local1109 setInfoAndAllocateIfNeeded(&fwOutput, fwOutputShape, &result); in executeOperation()