Searched refs:bwOutput (Results 1 – 3 of 3) sorted by relevance
164 T* bwOutput = nullptr; in executeTyped() local168 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()202 bwOutput = bwOutputTransposed.data(); in executeTyped()290 bwOutputBatchPtr = bwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()445 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare() local446 bwOutput.dimensions.resize(3); in prepare()447 bwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()448 bwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()449 bwOutput.dimensions[2] = bwNumUnits; in prepare()450 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
1012 RunTimeOperandInfo& bwOutput = in executeOperation() local1014 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()
3660 * * 1: bwOutput.