diff options
author | A. Unique TensorFlower <gardener@tensorflow.org> | 2017-10-15 21:36:48 -0700 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2017-10-15 21:41:29 -0700 |
commit | ca8af1d0dbb605087a4f8ae076188f2b9a26b1ba (patch) | |
tree | fc2b16452cfaba938365c21ba9221ca03953c28a /tensorflow/contrib/nccl/BUILD | |
parent | e30246c49b353b9136f69caef23e7ba0e9df0f0e (diff) |
Replace NcclReduce/Broadcast ops during graph optimization so that we can generate gradients for Reduce/Broadcast.
Changing _NcclBroadcastRecv shape input to int32 so that the corresponding Const op is outputting to HostMem.
PiperOrigin-RevId: 172279684
Diffstat (limited to 'tensorflow/contrib/nccl/BUILD')
-rw-r--r-- | tensorflow/contrib/nccl/BUILD | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/nccl/BUILD b/tensorflow/contrib/nccl/BUILD index d6508362b8..5e7263ff62 100644 --- a/tensorflow/contrib/nccl/BUILD +++ b/tensorflow/contrib/nccl/BUILD @@ -71,10 +71,12 @@ tf_kernel_library( "kernels/nccl_manager.cc", "kernels/nccl_manager.h", "kernels/nccl_ops.cc", + "kernels/nccl_rewrite.cc", ], deps = [ "//tensorflow/core:framework", "//tensorflow/core:gpu_headers_lib", + "//tensorflow/core:proto_text", "@nccl_archive//:nccl", ], alwayslink = 1, |