Searched refs:set_stage (Results 1 – 8 of 8) sorted by relevance
217 user_node->set_stage(graph->stage()); in Coloring()220 user_node_graph->set_stage(graph->stage()); in Coloring()253 user_node->set_stage(stage); in BroadCastColoring()472 parameter->set_stage(graph->stage()); in ParameterColoring()921 fg->set_stage(-1); in ElimGraphStage()
153 parallel_strategys->set_stage(UlongToUint(LongToUlong(node_stra.second->GetInputStage()))); in Save()
136 def set_stage(self, stage): member in Primitive
185 target_func_graph->set_stage(func_graph->stage()); in CloneFuncGraphValueNodes()767 new_func_graph->set_stage(func_graph->stage()); in TransformableClone()
400 void set_stage(int64_t stage) { stage_ = stage; } in set_stage() function
204 void set_stage(const int &stage) { stage_ = stage; } in set_stage() function
53 k_graph_->set_stage(primal_graph->stage()); in DFunctor()59 tape_->set_stage(primal_graph->stage()); in DFunctor()
333 func_graph->set_stage(stage); in ConvertCellObjToFuncGraph()