diff options
author | Ayush Dubey <ayushd@google.com> | 2018-08-30 19:49:48 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-30 19:54:26 -0700 |
commit | 73a3477356990f2451e220f553c9d7782df836ac (patch) | |
tree | 8894ebceebe247e59c0fbdb35ce92b45f63b6cdf /tensorflow/core/BUILD | |
parent | 1cb6544826551524f0f53f3e9632f71e67ea7851 (diff) |
Initialize collective_graph_key based on the graph if unspecified in RunOptions.
Before this CL, for collective_ops to work, the client had to specify a
collective_graph_key in the RunOptions of a session.Run call.
After this change, if a client does not specify a collective_graph_key for a
graph that contains collective ops, a graph key is generated automatically as a
hash of the set of keys of collective instances in the placed graph.
PiperOrigin-RevId: 211024617
Diffstat (limited to 'tensorflow/core/BUILD')
-rw-r--r-- | tensorflow/core/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/core/BUILD b/tensorflow/core/BUILD index 84b11024fd..92daafcd85 100644 --- a/tensorflow/core/BUILD +++ b/tensorflow/core/BUILD @@ -4078,6 +4078,7 @@ tf_cuda_cc_test( ":testlib", "//third_party/eigen3", "//tensorflow/cc:cc_ops", + "//tensorflow/core/kernels:collective_ops", "//tensorflow/core/kernels:control_flow_ops", "//tensorflow/core/kernels:cwise_op", "//tensorflow/core/kernels:dense_update_ops", @@ -4119,6 +4120,7 @@ tf_cc_test( "//tensorflow/cc:cc_ops", # Link with support for TensorFlow Debugger (tfdbg). "//tensorflow/core/debug", + "//tensorflow/core/kernels:collective_ops", "//tensorflow/core/kernels:control_flow_ops", "//tensorflow/core/kernels:cwise_op", "//tensorflow/core/kernels:dense_update_ops", |