| 31 | namespace grappler { |
| 32 | |
| 33 | GrapplerItem GrapplerItem::WithGraph(GraphDef&& graph_def) const { |
| 34 | GrapplerItem item; |
| 35 | item.id = id; |
| 36 | item.feed = feed; |
| 37 | item.fetch = fetch; |
| 38 | item.init_ops = init_ops; |
| 39 | item.keep_ops = keep_ops; |
| 40 | item.expected_init_time = expected_init_time; |
| 41 | item.save_op = save_op; |
| 42 | item.restore_op = restore_op; |
| 43 | item.save_restore_loc_tensor = save_restore_loc_tensor; |
| 44 | item.queue_runners = queue_runners; |
| 45 | item.devices_ = devices_; |
| 46 | item.optimization_options_ = optimization_options_; |
| 47 | item.graph.Swap(&graph_def); |
| 48 | return item; |
| 49 | } |
| 50 | |
| 51 | std::vector<const NodeDef*> GrapplerItem::MainOpsFanin() const { |
| 52 | return ComputeTransitiveFanin(graph, fetch); |