Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.h82 void recordTempAsSubModelOutput(uint32_t fromModelIndex) { in recordTempAsSubModelOutput() argument
83 const auto it = mOperandMap.find(fromModelIndex); in recordTempAsSubModelOutput()
85 mTempsAsSubModelOutputs.insert(std::make_pair(fromModelIndex, it->second)); in recordTempAsSubModelOutput()
230 void recordTemporaryDef(uint32_t fromModelIndex, uint32_t stepIndex) { in recordTemporaryDef() argument
232 nnAssert(temporaryToDefiningStep.count(fromModelIndex) == 0); in recordTemporaryDef()
233 temporaryToDefiningStep.insert(std::make_pair(fromModelIndex, stepIndex)); in recordTemporaryDef()
DExecutionPlan.cpp425 const uint32_t fromModelIndex = input.first; in findTempsAsSubModelOutputs() local
426 const auto it = mTemporaryToDefiningStep.find(fromModelIndex); in findTempsAsSubModelOutputs()
430 mSteps[stepIndex]->recordTempAsSubModelOutput(fromModelIndex); in findTempsAsSubModelOutputs()