diff options
author | 2017-10-30 07:22:44 -0700 | |
---|---|---|
committer | 2017-10-30 07:28:55 -0700 | |
commit | ef4490f637e17f3ce599f55522e63d06f470e540 (patch) | |
tree | 84d51477d3fb3f76147f6fb70be99d51745a7bd1 /tensorflow/contrib/nccl | |
parent | 2e54fd6de78d84af6b26f537ee25c5a625adce3b (diff) |
BUILD cleanup in contrib/...
PiperOrigin-RevId: 173889798
Diffstat (limited to 'tensorflow/contrib/nccl')
-rw-r--r-- | tensorflow/contrib/nccl/BUILD | 5 |
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/nccl/BUILD b/tensorflow/contrib/nccl/BUILD index 3aa3215a5f..ed9fb64b95 100644 --- a/tensorflow/contrib/nccl/BUILD +++ b/tensorflow/contrib/nccl/BUILD @@ -79,6 +79,7 @@ tf_kernel_library( "//tensorflow/core:gpu_headers_lib", "//tensorflow/core:lib", "//tensorflow/core:proto_text", + "//tensorflow/core:stream_executor", "@nccl_archive//:nccl", ], alwayslink = 1, @@ -114,7 +115,11 @@ tf_custom_op_py_library( deps = [ ":nccl_ops", "//tensorflow/contrib/util:util_py", + "//tensorflow/python:device", + "//tensorflow/python:framework_ops", "//tensorflow/python:platform", + "//tensorflow/python:util", + "//tensorflow/python/eager:context", ], ) |