diff options
author | 2017-07-27 15:52:47 -0700 | |
---|---|---|
committer | 2017-07-27 15:56:46 -0700 | |
commit | 6b7314de49affa69e64231ccac036d1baa411363 (patch) | |
tree | 1610918b6ff5a21ee54a8d97558b2c66421e489c /tensorflow/core/graph/graph_partition.h | |
parent | 28373cfe70dbb69031295fb3254e56f8b765b229 (diff) |
Consolidating the code to fill the partition's function library
into one place. Previously, Partition() and MasterSession::RegisterPartition()
both fills in the partitioned graph's function library.
PiperOrigin-RevId: 163400992
Diffstat (limited to 'tensorflow/core/graph/graph_partition.h')
-rw-r--r-- | tensorflow/core/graph/graph_partition.h | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/core/graph/graph_partition.h b/tensorflow/core/graph/graph_partition.h index 8820b8821f..67fafddd51 100644 --- a/tensorflow/core/graph/graph_partition.h +++ b/tensorflow/core/graph/graph_partition.h @@ -21,6 +21,7 @@ limitations under the License. #include <unordered_map> #include <vector> +#include "tensorflow/core/framework/function.h" #include "tensorflow/core/framework/graph.pb.h" #include "tensorflow/core/graph/costmodel.h" #include "tensorflow/core/graph/graph.h" @@ -45,6 +46,10 @@ struct PartitionOptions { typedef std::function<uint64(const string&)> GetIncarnationFunc; GetIncarnationFunc get_incarnation = nullptr; + // If specified, flib_def defines a function library that should be + // partitioned and replicated into each resulting partition graphs. + const FunctionLibraryDefinition* flib_def = nullptr; + // True if all the control flow "code" has already been added. The // control flow code needs to be added when we still have the entire // graph before any partitioning. So this flag should be false for |