Lines Matching refs:step

423     for (const auto& step : mSteps) {  in findTempsAsSubModelOutputs()  local
424 for (const auto& input : step->getTempsAsSubModelInputs()) { in findTempsAsSubModelOutputs()
594 for (const auto& step : mSteps) { in finish() local
595 int n = step->finishSubModel(fromModel, &mHasSubModelOutputOfUnknownSize, in finish()
655 for (const auto& step : compound()->mSteps) { in makeBursts() local
656 if (const auto preparedModel = step->getPreparedSubModel()) { in makeBursts()
710 for (const auto& step : compound()->mSteps) { in makeController() local
711 for (const auto& output: step->getTempsAsSubModelOutputs()) { in makeController()
815 const auto step = compoundBody->mSteps[controller->mNextStepIndex]; in next() local
816 *executor = std::make_shared<StepExecutor>(controller->mExecutionBuilder, step->getSubModel(), in next()
817 step->getDevice(), step->getPreparedSubModel()); in next()
818 (*executor)->setExecutionStep(step); in next()
819 step->mapInputsAndOutputs(*executor); in next()
827 const size_t firstSubModelOutputIndex = step->getModelOutputs().size(); in next()
828 const auto& subModelOutputs = step->getTempsAsSubModelOutputs(); in next()
848 const size_t firstSubModelInputIndex = step->getModelInputs().size(); in next()
849 const auto& subModelInputs = step->getTempsAsSubModelInputs(); in next()
871 step->getModelInputs().size() + step->getTempsAsSubModelInputs().size(); in next()
873 step->getOutputsAsSubModelInputsIndexToFromModel(); in next()
891 auto step = std::make_shared<ExecutionStep>(this, steps.size(), device); in createNewStep() local
892 steps.push_back(step); in createNewStep()
893 return step; in createNewStep()
959 for (const auto& step : mSteps) { in dump() local
960 step->dump(); in dump()
1029 std::shared_ptr<ExecutionStep> step = plan->createNewStep(devices[deviceIndex]); in partitionTheWork() local
1034 int n = step->addOperation(operationIndex, *this); in partitionTheWork()