Home
last modified time | relevance | path

Searched refs:fromModelIndex (Results 1 – 2 of 2) sorted by relevance

/frameworks/ml/nn/runtime/
DExecutionPlan.h74 void recordTempAsSubModelOutput(uint32_t fromModelIndex) { in recordTempAsSubModelOutput() argument
75 const auto it = mOperandMap.find(fromModelIndex); in recordTempAsSubModelOutput()
77 mTempsAsSubModelOutputs.insert(std::make_pair(fromModelIndex, it->second)); in recordTempAsSubModelOutput()
209 void recordTemporaryDef(uint32_t fromModelIndex, uint32_t stepIndex) { in recordTemporaryDef() argument
211 nnAssert(temporaryToDefiningStep.count(fromModelIndex) == 0); in recordTemporaryDef()
212 temporaryToDefiningStep.insert(std::make_pair(fromModelIndex, stepIndex)); in recordTemporaryDef()
DExecutionPlan.cpp281 const uint32_t fromModelIndex = input.first; in findTempsAsSubModelOutputs() local
282 const auto it = mTemporaryToDefiningStep.find(fromModelIndex); in findTempsAsSubModelOutputs()
286 mSteps[stepIndex]->recordTempAsSubModelOutput(fromModelIndex); in findTempsAsSubModelOutputs()