| 41 | namespace TEngine { |
| 42 | |
| 43 | void DevAllocator::SameGraph(Graph* graph, DevExecutor* dev_executor, std::vector<Subgraph*>& sub_list) |
| 44 | { |
| 45 | Subgraph* sub_graph = new Subgraph(graph->GetName()); |
| 46 | |
| 47 | sub_graph->seq_nodes = graph->seq_nodes; |
| 48 | sub_graph->input_nodes = graph->input_nodes; |
| 49 | sub_graph->output_nodes = graph->output_nodes; |
| 50 | |
| 51 | sub_graph->SetAttr("dev_executor", dev_executor); |
| 52 | sub_list.push_back(sub_graph); |
| 53 | } |
| 54 | |
| 55 | void DevAllocator::PartitionGraph(GenericEngine* engine, GraphExecutor* graph_executor, Graph* graph, |
| 56 | std::vector<Subgraph*>& sub_list) |