Home
last modified time | relevance | path

Searched refs:sourceModelIndex (Results 1 – 4 of 4) sorted by relevance

/aosp12/packages/modules/NeuralNetworks/runtime/
H A DExecutionPlan.cpp408 mSourceModelIndex(sourceModelIndex), in ExecutionStep()
1853 ExecutionStep* ExecutionPlan::createNewExecutionStep(uint32_t sourceModelIndex, in createNewExecutionStep() argument
2071 const uint32_t sourceModelIndex = step->getSourceModelIndex(); in forEachDynamicTemporary() local
2087 uint32_t sourceModelIndex = plan->getSourceModels().addModel(this); in partitionTheWork() local
2099 int ModelBuilder::partitionTheWorkInternal(uint32_t sourceModelIndex, in partitionTheWorkInternal() argument
2112 << "sourceModelIndex = " << sourceModelIndex << ", " in partitionTheWorkInternal()
2130 if (sourceModelIndex == kMainModelInSourceModels && in partitionTheWorkInternal()
2251 ifStep->outerInputOperands.emplace_back(sourceModelIndex, in partitionTheWorkInternal()
2255 ifStep->outerOutputOperands.emplace_back(sourceModelIndex, in partitionTheWorkInternal()
2312 whileStep->outerInputOperands.emplace_back(sourceModelIndex, in partitionTheWorkInternal()
[all …]
H A DModelBuilder.h161 int partitionTheWorkInternal(uint32_t sourceModelIndex,
H A DExecutionPlan.h238 ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex,
703 ExecutionStep* createNewExecutionStep(uint32_t sourceModelIndex,
H A DExecutionBuilder.cpp1232 const uint32_t sourceModelIndex = mExecutionStep->getSourceModelIndex(); in updateOutputShapes() local
1240 const auto sourceOperandIndex = SourceOperandIndex(sourceModelIndex, it->second); in updateOutputShapes()