Searched refs:stepIndex (Results 1 – 2 of 2) sorted by relevance
| /frameworks/ml/nn/runtime/ |
| D | ExecutionPlan.h | 54 ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, std::shared_ptr<Device> device);
230 void recordTemporaryDef(uint32_t fromModelIndex, uint32_t stepIndex) { in recordTemporaryDef() argument
233 temporaryToDefiningStep.insert(std::make_pair(fromModelIndex, stepIndex)); in recordTemporaryDef()
|
| D | ExecutionPlan.cpp | 265 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, in ExecutionStep() argument
267 : mPlan(plan), mIndex(stepIndex), mSubModel(), mDevice(device), mToken(plan->getCacheToken()) {} in ExecutionStep()
428 const uint32_t stepIndex = it->second; in findTempsAsSubModelOutputs() local
429 nnAssert(stepIndex < mSteps.size()); in findTempsAsSubModelOutputs()
430 mSteps[stepIndex]->recordTempAsSubModelOutput(fromModelIndex); in findTempsAsSubModelOutputs()
|