|
|
|
@@ -1110,10 +1110,11 @@ bool ExecutorPy::AddDFGraph(const py::dict& init_params, const std::string& phas |
|
|
|
(void)DfGraphManager::GetInstance().AddGraph(phase, convertor.GetComputeGraph()); |
|
|
|
} |
|
|
|
(void)DfGraphManager::GetInstance().AddGraph(init_graph, convertor.GetInitGraph()); |
|
|
|
(void)DfGraphManager::GetInstance().AddGraph(checkpoint_name, convertor.GetSaveCheckpointGraph()); |
|
|
|
(void)DfGraphManager::GetInstance().AddGraph(BROADCAST_GRAPH_NAME, convertor.GetBroadcastGraph()); |
|
|
|
|
|
|
|
DfGraphManager::GetInstance().SetAnfGraph(checkpoint_name, anf_graph); |
|
|
|
Status ret = DfGraphManager::GetInstance().AddGraph(checkpoint_name, convertor.GetSaveCheckpointGraph()); |
|
|
|
if (ret == Status::SUCCESS) { |
|
|
|
DfGraphManager::GetInstance().SetAnfGraph(checkpoint_name, anf_graph); |
|
|
|
} |
|
|
|
|
|
|
|
return true; |
|
|
|
} |
|
|
|
|