Searched refs:bwOutput (Results 1 – 3 of 3) sorted by relevance
166 T* bwOutput = nullptr; in executeTyped() local170 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()204 bwOutput = bwOutputTransposed.data(); in executeTyped()292 bwOutputBatchPtr = bwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()446 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare() local447 bwOutput.dimensions.resize(3); in prepare()448 bwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()449 bwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()450 bwOutput.dimensions[2] = bwNumUnits; in prepare()451 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
1068 RunTimeOperandInfo& bwOutput = in executeOperation() local1070 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()
3554 * * 1: bwOutput.