diff options
author | 2018-07-27 14:58:10 -0700 | |
---|---|---|
committer | 2018-07-27 15:02:45 -0700 | |
commit | 2beb3a9d8b9df294e7635cc23d195a76fd78de79 (patch) | |
tree | c53e0ba8f738093c5e97e5bc2ee6a5b064afcf2b /tensorflow/python/distribute/BUILD | |
parent | 470b43af3153942e4ef838610aa07e93f904fd5f (diff) |
Add distribute_coordinator: a unified and split client for distributed traning.
PiperOrigin-RevId: 206378953
Diffstat (limited to 'tensorflow/python/distribute/BUILD')
-rw-r--r-- | tensorflow/python/distribute/BUILD | 40 |
1 files changed, 40 insertions, 0 deletions
diff --git a/tensorflow/python/distribute/BUILD b/tensorflow/python/distribute/BUILD new file mode 100644 index 0000000000..a29043d8b8 --- /dev/null +++ b/tensorflow/python/distribute/BUILD @@ -0,0 +1,40 @@ +licenses(["notice"]) # Apache 2.0 + +package( + default_visibility = [ + "//tensorflow:internal", + ], +) + +py_library( + name = "distribute_coordinator", + srcs = [ + "distribute_coordinator.py", + ], + srcs_version = "PY2AND3", + deps = [ + "//tensorflow/core:protos_all_py", + "//tensorflow/python:training", + ], +) + +py_test( + name = "distribute_coordinator_test", + size = "small", + srcs = ["distribute_coordinator_test.py"], + srcs_version = "PY2AND3", + deps = [ + ":distribute_coordinator", + "//tensorflow/core:protos_all_py", + "//tensorflow/python:client_testlib", + "//tensorflow/python:control_flow_ops", + "//tensorflow/python:distributed_framework_test_lib", + "//tensorflow/python:framework_ops", + "//tensorflow/python:framework_test_lib", + "//tensorflow/python:math_ops", + "//tensorflow/python:session", + "//tensorflow/python:training", + "//tensorflow/python:variable_scope", + "//tensorflow/python:variables", + ], +) |