Lines Matching refs:mOutputs
168 mOutputs(mModel->outputCount()) { in ExecutionBuilder()
237 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in setOutput()
251 return mOutputs[index].setFromPointer(mModel->getOutputOperand(index), type, buffer, l); in setOutput()
263 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in setOutputFromMemory()
283 return mOutputs[index].setFromMemory(mModel->getOutputOperand(index), type, poolIndex, offset, in setOutputFromMemory()
342 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in getOutputOperandDimensions()
348 const auto& dims = mOutputs[index].dimensions; in getOutputOperandDimensions()
355 return mOutputs[index].isSufficient ? ANEURALNETWORKS_NO_ERROR in getOutputOperandDimensions()
365 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in getOutputOperandRank()
371 *rank = static_cast<uint32_t>(mOutputs[index].dimensions.size()); in getOutputOperandRank()
372 return mOutputs[index].isSufficient ? ANEURALNETWORKS_NO_ERROR in getOutputOperandRank()
552 for (auto& p : mOutputs) { in compute()
606 outputShapes->resize(mOutputs.size()); in initializeOutputShapes()
607 for (uint32_t i = 0; i < mOutputs.size(); i++) { in initializeOutputShapes()
608 (*outputShapes)[i].dimensions = mOutputs[i].dimensions; in initializeOutputShapes()
628 NN_RET_CHECK_EQ(outputShapes.size(), mOutputs.size()); in updateOutputShapes()
631 NN_RET_CHECK(isUpdatable(mOutputs[i].dimensions, outputShapes[i].dimensions)); in updateOutputShapes()
634 mOutputs[i].dimensions = outputShapes[i].dimensions; in updateOutputShapes()
635 mOutputs[i].isSufficient = outputShapes[i].isSufficient; in updateOutputShapes()
724 mOutputs(model->outputCount()) { in StepExecutor()
730 mOutputs = mExecutionBuilder->mOutputs; in mapInputsAndOutputsTrivially()
802 logArguments("output", mOutputs); in startCompute()
857 n = allocatePointerArgumentsToPool(&mOutputs, &outputPointerArguments); in startComputeOnDevice()
879 setRequestArgumentArray(mOutputs, &request.outputs); in startComputeOnDevice()
978 for (auto& info : mOutputs) { in startComputeOnDevice()
1049 fixPointerArguments(mOutputs); in startComputeOnCpu()
1053 setRequestArgumentArray(mOutputs, &request.outputs); in startComputeOnCpu()