Searched refs:bwOutput (Results 1 – 2 of 2) sorted by relevance
115 T* bwOutput = nullptr; in executeTyped() local119 bwOutput = context->getOutputBuffer<T>(kBwOutputTensor); in executeTyped()153 bwOutput = bwOutputTransposed.data(); in executeTyped()222 bwOutputBatchPtr = bwOutput + i * batchSize * bwOutputBatchStride; in executeTyped()367 Shape bwOutput = context->getOutputShape(kBwOutputTensor); in prepare() local368 bwOutput.dimensions.resize(3); in prepare()369 bwOutput.dimensions[0] = timeMajor ? maxTime : batchSize; in prepare()370 bwOutput.dimensions[1] = timeMajor ? batchSize : maxTime; in prepare()371 bwOutput.dimensions[2] = bwNumUnits; in prepare()372 NN_RET_CHECK(context->setOutputShape(kBwOutputTensor, bwOutput)); in prepare()
1111 RunTimeOperandInfo& bwOutput = in executeOperation() local1113 success = success && setInfoAndAllocateIfNeeded(&bwOutput, bwOutputShape, &result); in executeOperation()