Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.cpp1448 step->ifStep()->index = compound()->mSteps.size(); in createNewIfStep()
1450 return step->ifStep(); in createNewIfStep()
1742 IfStep* ifStep = plan->createNewIfStep(); in partitionTheWorkInternal() local
1743 ifStep->conditionOperandIndex = SourceOperandIndex( in partitionTheWorkInternal()
1745 ifStep->thenStepIndex = plan->getNextStepIndex(); in partitionTheWorkInternal()
1749 ifStep->elseStepIndex = plan->getNextStepIndex(); in partitionTheWorkInternal()
1756 ifStep->outerInputOperands.emplace_back(sourceModelIndex, in partitionTheWorkInternal()
1760 ifStep->outerOutputOperands.emplace_back(sourceModelIndex, in partitionTheWorkInternal()
1765 ifStep->thenBranchInputOperands.emplace_back( in partitionTheWorkInternal()
1769 ifStep->thenBranchOutputOperands.emplace_back( in partitionTheWorkInternal()
[all …]
DExecutionPlan.h356 IfStep* ifStep() { return &std::get<IfStep>(mStep); } in ifStep() function
362 const IfStep* ifStep() const { return &std::get<IfStep>(mStep); } in ifStep() function