Searched refs:sourceModelIndex (Results 1 – 4 of 4) sorted by relevance
| /packages/modules/NeuralNetworks/runtime/ |
| D | ExecutionPlan.cpp | 404 ExecutionStep::ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex, in ExecutionStep() argument
408 mSourceModelIndex(sourceModelIndex), in ExecutionStep()
1853 ExecutionStep* ExecutionPlan::createNewExecutionStep(uint32_t sourceModelIndex, in createNewExecutionStep() argument
1857 compound()->mSteps.size(), sourceModelIndex, device); in createNewExecutionStep()
2071 const uint32_t sourceModelIndex = step->getSourceModelIndex(); in forEachDynamicTemporary() local
2073 const auto sourceOperandIndex = SourceOperandIndex(sourceModelIndex, entry.first); in forEachDynamicTemporary()
2087 uint32_t sourceModelIndex = plan->getSourceModels().addModel(this); in partitionTheWork() local
2088 NN_RETURN_IF_ERROR(partitionTheWorkInternal(sourceModelIndex, devices, preference, priority, in partitionTheWork()
2099 int ModelBuilder::partitionTheWorkInternal(uint32_t sourceModelIndex, in partitionTheWorkInternal() argument
2112 << "sourceModelIndex = " << sourceModelIndex << ", " in partitionTheWorkInternal()
[all …]
|
| D | ModelBuilder.h | 161 int partitionTheWorkInternal(uint32_t sourceModelIndex,
|
| D | ExecutionPlan.h | 238 ExecutionStep(ExecutionPlan* plan, uint32_t stepIndex, uint32_t sourceModelIndex,
703 ExecutionStep* createNewExecutionStep(uint32_t sourceModelIndex,
|
| D | ExecutionBuilder.cpp | 1232 const uint32_t sourceModelIndex = mExecutionStep->getSourceModelIndex(); in updateOutputShapes() local
1240 const auto sourceOperandIndex = SourceOperandIndex(sourceModelIndex, it->second); in updateOutputShapes()
|