Lines Matching refs:mOutputs

98       mOutputs(mModel->outputCount()) {  in ExecutionBuilder()
100 << " inputs and " << mOutputs.size() << " outputs"; in ExecutionBuilder()
215 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in setOutput()
229 if (!mOutputs[index].unspecified()) { in setOutput()
235 std::tie(n, mOutputs[index]) = in setOutput()
249 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in setOutputFromMemory()
273 if (!mOutputs[index].unspecified()) { in setOutputFromMemory()
279 std::tie(n, mOutputs[index]) = ModelArgumentInfo::createFromMemory( in setOutputFromMemory()
416 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in getOutputOperandDimensions()
422 const auto& dims = mOutputs[index].dimensions(); in getOutputOperandDimensions()
429 return mOutputs[index].isSufficient() ? ANEURALNETWORKS_NO_ERROR in getOutputOperandDimensions()
444 uint32_t count = static_cast<uint32_t>(mOutputs.size()); in getOutputOperandRank()
450 *rank = static_cast<uint32_t>(mOutputs[index].dimensions().size()); in getOutputOperandRank()
451 return mOutputs[index].isSufficient() ? ANEURALNETWORKS_NO_ERROR in getOutputOperandRank()
755 for (auto& p : mOutputs) { in computeFenced()
762 for (uint32_t i = 0; i < mOutputs.size(); i++) { in computeFenced()
763 if (mOutputs[i].state() != ModelArgumentInfo::HAS_NO_VALUE && in computeFenced()
816 for (auto& p : mOutputs) { in compute()
876 std::vector<OutputShape> outputShapes(mOutputs.size()); in getInitialOutputShapes()
877 std::transform(mOutputs.begin(), mOutputs.end(), outputShapes.begin(), in getInitialOutputShapes()
903 NN_RET_CHECK_EQ(outputShapes.size(), mOutputs.size()); in updateOutputShapes()
906 NN_RET_CHECK(isUpdatable(mOutputs[i].dimensions(), outputShapes[i].dimensions)); in updateOutputShapes()
912 mOutputs[i].dimensions() = outputShapes[i].dimensions; in updateOutputShapes()
913 mOutputs[i].isSufficient() = outputShapes[i].isSufficient; in updateOutputShapes()
919 for (const auto& output : mOutputs) { in updateMemories()
936 for (const auto& output : mOutputs) { in finishWithoutSyncFence()
989 mOutputs(model->outputCount()) { in StepExecutor()
992 << mOutputs.size() << " outputs"; in StepExecutor()
997 mOutputs = mExecutionBuilder->mOutputs; in mapInputsAndOutputsTrivially()
1088 logArguments("output", mOutputs); in computeWithMemories()
1095 mInputs, mOutputs, memories, burstController, measure, deadline, loopTimeoutDuration); in computeWithMemories()
1108 logArguments("output", mOutputs); in computeFenced()
1119 mInputs, mOutputs, mMemories.getObjects(), waitFor, measure, deadline, in computeFenced()
1159 for (auto& output : mOutputs) { in computeOnCpuFallback()