Searched defs:stepIndex (Results 1 - 2 of 2) sorted by relevance

/frameworks/ml/nn/runtime/
H A DExecutionPlan.h48 uint32_t stepIndex,
209 void recordTemporaryDef(uint32_t fromModelIndex, uint32_t stepIndex) { argument
212 temporaryToDefiningStep.insert(std::make_pair(fromModelIndex, stepIndex));
H A DExecutionPlan.cpp131 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, argument
133 : mPlan(plan), mIndex(stepIndex), mSubModel(), mDevice(device) {}
284 const uint32_t stepIndex = it->second; local
285 nnAssert(stepIndex < mSteps.size());
286 mSteps[stepIndex]->recordTempAsSubModelOutput(fromModelIndex);

Completed in 63 milliseconds