Skip to content
Snippets Groups Projects
Commit 6dcedc11 authored by Derek Murray's avatar Derek Murray Committed by TensorFlower Gardener
Browse files

Add new overload of `GraphToFunctionDef()` that can consume the graph.

Previously, we copied all of the node data from the input subgraph to the `FunctionDef`. This change reduces the memory footprint of multi-device function instantiation by (where possible) `std::move`-ing the bulk of the node data from the input graph to newly created `NodeDef` members of the resulting `FunctionDef` protocol buffer.

PiperOrigin-RevId: 617641191
parent 3dd43f8f
No related merge requests found
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment