Lines Matching refs:sourceModelIndex
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()
2130 if (sourceModelIndex == kMainModelInSourceModels && in partitionTheWorkInternal()
2203 plan->createNewExecutionStep(sourceModelIndex, devices[deviceIndex]); in partitionTheWorkInternal()
2239 sourceModelIndex, operation.inputs[op::kCondBoolOperand]); 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()
2316 whileStep->outerOutputOperands.emplace_back(sourceModelIndex, in partitionTheWorkInternal()