Home
last modified time | relevance | path

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

/packages/modules/NeuralNetworks/runtime/
DCompilationBuilder.cpp69 mPlan.setCaching(&mCacheInfo, mToken); in finish()
72 int n = mModel->partitionTheWork(mDevices, mPreference, mPriority, deadline, &mPlan, in finish()
105 mPlan.reset(); in finish()
106 mPlan.becomeSingleStep(DeviceManager::getCpuDevice(), mModel); in finish()
107 return mPlan.finish(mPreference, mPriority, deadline, mMetadata, ANEURALNETWORKS_NO_ERROR); in finish()
286 if (!mPlan.isValid()) { in getPreferredMemoryAlignmentForInput()
297 *alignment = mPlan.getMemoryPreference(IOType::INPUT, index).alignment; in getPreferredMemoryAlignmentForInput()
308 if (!mPlan.isValid()) { in getPreferredMemoryPaddingForInput()
319 *padding = mPlan.getMemoryPreference(IOType::INPUT, index).padding; in getPreferredMemoryPaddingForInput()
331 if (!mPlan.isValid()) { in getPreferredMemoryAlignmentForOutput()
[all …]
DCompilationBuilder.h88 bool hasDynamicTemporaries() const { return mPlan.hasDynamicTemporaries(); } in hasDynamicTemporaries()
94 const ExecutionPlan& forTest_getExecutionPlan() const { return mPlan; } in forTest_getExecutionPlan()
108 ExecutionPlan mPlan; variable
DExecutionBuilder.cpp156 mPlan(&compilation->mPlan), in ExecutionBuilder()
166 CHECK(mPlan->isSimple()); in SimpleExecutionBuilder()
171 CHECK(mPlan->isCompound()); in CompoundExecutionBuilder()
175 return mPlan->getSourceModels().getModel(index); in getSourceModel()
688 mExecutor = mPlan->makeStepExecutor(mReusable, this); in computeInternal()
723 auto controller = mPlan->makeController(this, burstBuilder); in computeInternal()
737 ? mPlan->fallback(controller, &executor, &burstController, &outputShapes) in computeInternal()
738 : mPlan->next(controller, &executor, &burstController, &outputShapes); in computeInternal()
813 cpuFallbackPartial(*mPlan, controller); in computeInternal()
892 mExecutor = mPlan->makeStepExecutor(mReusable, this); in computeFencedInternal()
[all …]
DExecutionPlan.h324 ExecutionPlan* mPlan; variable
630 [[maybe_unused]] const ExecutionPlan* mPlan; variable
873 CompoundBody(const ExecutionPlan* plan) : mPlan(plan) { CHECK(plan != nullptr); } in CompoundBody()
969 const ExecutionPlan* mPlan; member
DExecutionPlan.cpp446 : mPlan(plan), in ExecutionStep()
516 mPlan->recordTemporaryDef(SourceOperandIndex(mSourceModelIndex, sourceOperandIndex), in addOperand()
535 mPlan->recordOutputDef(SourceOperandIndex(mSourceModelIndex, sourceOperandIndex), in addOperand()
733 return mPlan->getSourceModels().getModel(mSourceModelIndex); in getSourceModel()
875 return compile(*mDevice, mStepModel, executionPreference, priority, {}, *mPlan->getCacheInfo(), in finishStepModel()
1179 : mPlan(plan), in Controller()
2068 const auto sourceOperandIndex = mPlan->getInputSourceOperand(index); in forEachStepRoleOfInput()
2075 const auto sourceOperandIndex = mPlan->getOutputSourceOperand(index); in forEachStepRoleOfOutput()
DExecutionBuilder.h189 const ExecutionPlan* mPlan; variable
/packages/modules/NeuralNetworks/runtime/test/
DTestPartitioning.cpp3396 ExecutePriority::DEFAULT, {}, &mPlan), in finishAndPartitionModelForDevices()
3403 mPlan.forEachStepRoleOfInput( in checkStepRolesOfInput()
3414 mPlan.forEachStepRoleOfOutput( in checkStepRolesOfOutput()
3426 mPlan.forTest_compoundForEachStepRoleOfSourceOperand( in checkStepRolesOfSourceOperand()
3435 ExecutionPlan mPlan; member in __anonb50294d30111::MemoryStepRoleTest
3456 checkExecutionPlanSteps(mPlan, {"deviceB", "deviceA", "deviceC"}); in TEST_F()
3506 checkExecutionPlanSteps(mPlan, {kIfStep, "deviceA", kGotoStep, "deviceB"}); in TEST_F()
3570 checkExecutionPlanSteps(mPlan, {kWhileStep, "deviceA", kGotoStep, "deviceB", kGotoStep}); in TEST_F()
3662 checkExecutionPlanSteps(mPlan, {kWhileStep, "deviceA", kGotoStep, kIfStep, "deviceB", kGotoStep, in TEST_F()