diff options
Diffstat (limited to 'tensorflow/core/common_runtime/build_graph_options.h')
| -rw-r--r-- | tensorflow/core/common_runtime/build_graph_options.h | 3 |
1 files changed, 3 insertions, 0 deletions
diff --git a/tensorflow/core/common_runtime/build_graph_options.h b/tensorflow/core/common_runtime/build_graph_options.h index 5ca170e922..3d0f242ea5 100644 --- a/tensorflow/core/common_runtime/build_graph_options.h +++ b/tensorflow/core/common_runtime/build_graph_options.h @@ -31,6 +31,9 @@ struct BuildGraphOptions { // TODO(mrry): Remove this when the distributed runtime supports Arg/Retval. bool use_function_convention = false; + static const int64 kNoCollectiveGraphKey = 0; + int64 collective_graph_key = kNoCollectiveGraphKey; + string DebugString() const; }; |