Home
last modified time | relevance | path

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

/frameworks/ml/nn/runtime/
DExecutionPlan.h301 struct WhileStep { struct
351 bool isWhile() const { return std::holds_alternative<WhileStep>(mStep); } in isWhile()
357 WhileStep* whileStep() { return &std::get<WhileStep>(mStep); } in whileStep()
363 const WhileStep* whileStep() const { return &std::get<WhileStep>(mStep); } in whileStep()
369 WhileStep* tryWhileStep() { return std::get_if<WhileStep>(&mStep); } in tryWhileStep()
375 const WhileStep* tryWhileStep() const { return std::get_if<WhileStep>(&mStep); } in tryWhileStep()
381 std::variant<ExecutionStep, IfStep, WhileStep, GotoStep> mStep;
385 std::string toString(const WhileStep& step);
530 WhileStep* createNewWhileStep();
623 int nextCompound(const WhileStep* step, std::shared_ptr<Controller> controller,
DExecutionPlan.cpp411 } else if (const WhileStep* step = logicalStep->tryWhileStep()) { in findTempsAsStepModelOutputs()
584 std::string toString(const WhileStep& step) { in toString()
601 } else if (const WhileStep* step = tryWhileStep()) { in dump()
648 } else if (WhileStep* step = logicalStep->tryWhileStep()) { in finish()
707 } else if (const WhileStep* step = logicalStep->tryWhileStep()) { in findControlFlowBoundaryConstants()
928 } else if (const WhileStep* step = logicalStep->tryWhileStep()) { in makeController()
1155 } else if (const WhileStep* step = logicalStep->tryWhileStep()) { in nextCompound()
1299 int ExecutionPlan::nextCompound(const WhileStep* step, std::shared_ptr<Controller> controller, in nextCompound()
1453 WhileStep* ExecutionPlan::createNewWhileStep() { in createNewWhileStep()
1455 auto step = std::make_shared<LogicalStep>(std::in_place_type<WhileStep>); in createNewWhileStep()
[all …]