Home
last modified time | relevance | path

Searched refs:mPlan (Results 1 – 7 of 7) sorted by relevance

/packages/modules/NeuralNetworks/runtime/
DCompilationBuilder.cpp64 mPlan.setCaching(&mCacheInfo, mToken); in finish()
67 int n = mModel->partitionTheWork(mDevices, mPreference, mPriority, deadline, &mPlan, in finish()
99 mPlan.reset(); in finish()
100 mPlan.becomeSingleStep(DeviceManager::getCpuDevice(), mModel); in finish()
101 return mPlan.finish(mPreference, mPriority, deadline, ANEURALNETWORKS_NO_ERROR); in finish()
253 if (!mPlan.isValid()) { in getPreferredMemoryAlignmentForInput()
264 *alignment = mPlan.getMemoryPreference(IOType::INPUT, index).alignment; in getPreferredMemoryAlignmentForInput()
275 if (!mPlan.isValid()) { in getPreferredMemoryPaddingForInput()
286 *padding = mPlan.getMemoryPreference(IOType::INPUT, index).padding; in getPreferredMemoryPaddingForInput()
298 if (!mPlan.isValid()) { in getPreferredMemoryAlignmentForOutput()
[all …]
DCompilationBuilder.h79 bool hasDynamicTemporaries() const { return mPlan.hasDynamicTemporaries(); } in hasDynamicTemporaries()
83 const ExecutionPlan& forTest_getExecutionPlan() const { return mPlan; } in forTest_getExecutionPlan()
91 ExecutionPlan mPlan; variable
DExecutionBuilder.cpp155 mPlan(&compilation->mPlan), in ExecutionBuilder()
165 CHECK(mPlan->isSimple()); in SimpleExecutionBuilder()
170 CHECK(mPlan->isCompound()); in CompoundExecutionBuilder()
174 return mPlan->getSourceModels().getModel(index); in getSourceModel()
649 mExecutor = mPlan->makeStepExecutor(mReusable, this); in computeInternal()
684 auto controller = mPlan->makeController(this, burstBuilder); in computeInternal()
698 ? mPlan->fallback(controller, &executor, &burstController, &outputShapes) in computeInternal()
699 : mPlan->next(controller, &executor, &burstController, &outputShapes); in computeInternal()
774 cpuFallbackPartial(*mPlan, controller); in computeInternal()
853 mExecutor = mPlan->makeStepExecutor(mReusable, this); in computeFencedInternal()
[all …]
DExecutionPlan.h324 ExecutionPlan* mPlan; variable
630 [[maybe_unused]] const ExecutionPlan* mPlan; variable
871 CompoundBody(const ExecutionPlan* plan) : mPlan(plan) { CHECK(plan != nullptr); } in CompoundBody()
966 const ExecutionPlan* mPlan; member
DExecutionPlan.cpp406 : mPlan(plan), in ExecutionStep()
476 mPlan->recordTemporaryDef(SourceOperandIndex(mSourceModelIndex, sourceOperandIndex), in addOperand()
495 mPlan->recordOutputDef(SourceOperandIndex(mSourceModelIndex, sourceOperandIndex), in addOperand()
693 return mPlan->getSourceModels().getModel(mSourceModelIndex); in getSourceModel()
835 return compile(*mDevice, mStepModel, executionPreference, priority, {}, *mPlan->getCacheInfo(), in finishStepModel()
1127 : mPlan(plan), in Controller()
2016 const auto sourceOperandIndex = mPlan->getInputSourceOperand(index); in forEachStepRoleOfInput()
2023 const auto sourceOperandIndex = mPlan->getOutputSourceOperand(index); in forEachStepRoleOfOutput()
DExecutionBuilder.h170 const ExecutionPlan* mPlan; variable
/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp3394 ExecutePriority::DEFAULT, {}, &mPlan), in finishAndPartitionModelForDevices()
3401 mPlan.forEachStepRoleOfInput( in checkStepRolesOfInput()
3412 mPlan.forEachStepRoleOfOutput( in checkStepRolesOfOutput()
3424 mPlan.forTest_compoundForEachStepRoleOfSourceOperand( in checkStepRolesOfSourceOperand()
3433 ExecutionPlan mPlan; member in __anon27781e9a0111::MemoryStepRoleTest
3454 checkExecutionPlanSteps(mPlan, {"deviceB", "deviceA", "deviceC"}); in TEST_F()
3504 checkExecutionPlanSteps(mPlan, {kIfStep, "deviceA", kGotoStep, "deviceB"}); in TEST_F()
3568 checkExecutionPlanSteps(mPlan, {kWhileStep, "deviceA", kGotoStep, "deviceB", kGotoStep}); in TEST_F()
3660 checkExecutionPlanSteps(mPlan, {kWhileStep, "deviceA", kGotoStep, kIfStep, "deviceB", kGotoStep, in TEST_F()