Searched refs:bwOutput (Results 1 – 3 of 3) sorted by relevance
135 T* bwOutput = nullptr; in executeTyped() local139 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()173 bwOutput = bwOutputTransposed.data(); in executeTyped()261 bwOutputBatchPtr = bwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()386 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare() local387 bwOutput.dimensions.resize(3); in prepare()388 bwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()389 bwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()390 bwOutput.dimensions[2] = bwNumUnits; in prepare()391 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
1013 RunTimeOperandInfo& bwOutput = in executeOperation() local1015 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()
3717 * * 1: bwOutput.