experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Expose distribute/ packages inside the v2 pip package.

Browse Source 
PiperOrigin-RevId: 248377124
This commit is contained in:
Igor Saprykin 2019-05-15 12:00:21 -07:00 committed by TensorFlower Gardener
parent 3bad08e34e
commit 6700fd4d24
2 changed files with 16 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

24
tensorflow/python/distribute/BUILD
View File

@ -11,6 +11,21 @@ load("//tensorflow:tensorflow.bzl", "cuda_py_test")
load("//tensorflow/compiler/tests:build_defs.bzl", "tf_xla_py_test") load("//tensorflow/compiler/tests:build_defs.bzl", "tf_xla_py_test")
load("//tensorflow/core:platform/default/distribute.bzl", "distribute_py_test") load("//tensorflow/core:platform/default/distribute.bzl", "distribute_py_test")
py_library(
name = "distribute_test_lib_pip",
deps = [
":combinations",
":model_combinations",
":multi_worker_test_base",
":saved_model_test_base",
":single_loss_example",
":strategy_combinations",
":strategy_test_lib",
"//tensorflow/python/keras/distribute:keras_correctness_test_lib",
"//tensorflow/python/keras/distribute:keras_test_lib",
],
)
py_library( py_library(
name = "all_reduce", name = "all_reduce",
srcs = [ srcs = [
@ -201,9 +216,6 @@ py_test(
srcs = ["distribute_coordinator_test.py"], srcs = ["distribute_coordinator_test.py"],
python_version = "PY2", python_version = "PY2",
srcs_version = "PY2AND3", srcs_version = "PY2AND3",
tags = [
"no_pip", # b/131691139
],
deps = [ deps = [
":distribute_coordinator", ":distribute_coordinator",
"//tensorflow/core:protos_all_py", "//tensorflow/core:protos_all_py",
@ -710,9 +722,6 @@ cuda_py_test(
"//tensorflow/python/eager:test", "//tensorflow/python/eager:test",
], ],
grpc_enabled = True, grpc_enabled = True,
tags = [
"no_pip", # b/131691139
],
) )
py_library( py_library(
@ -813,7 +822,6 @@ distribute_py_test(
main = "minimize_loss_test.py", main = "minimize_loss_test.py",
tags = [ tags = [
"multi_and_single_gpu", "multi_and_single_gpu",
"no_pip", # b/131691139
], ],
deps = [ deps = [
":mirrored_strategy", ":mirrored_strategy",
@ -866,7 +874,6 @@ distribute_py_test(
main = "step_fn_test.py", main = "step_fn_test.py",
tags = [ tags = [
"multi_and_single_gpu", "multi_and_single_gpu",
"no_pip", # b/131691139
], ],
deps = [ deps = [
":single_loss_example", ":single_loss_example",
@ -926,7 +933,6 @@ cuda_py_test(
tags = [ tags = [
"guitar", "guitar",
"multi_and_single_gpu", "multi_and_single_gpu",
"no_pip", # b/131691139
"no_windows_gpu", # TODO(b/130551176) "no_windows_gpu", # TODO(b/130551176)
], ],
) )

1
tensorflow/tools/pip_package/BUILD
View File

@ -70,6 +70,7 @@ COMMON_PIP_DEPS = [
"//tensorflow/python/compiler:compiler", "//tensorflow/python/compiler:compiler",
"//tensorflow/python:cond_v2", "//tensorflow/python:cond_v2",
"//tensorflow/python:distributed_framework_test_lib", "//tensorflow/python:distributed_framework_test_lib",
"//tensorflow/python/distribute:distribute_test_lib_pip",
"//tensorflow/python:loss_scale", "//tensorflow/python:loss_scale",
"//tensorflow/python:loss_scale_optimizer", "//tensorflow/python:loss_scale_optimizer",
"//tensorflow/python:meta_graph_testdata", "//tensorflow/python:meta_graph_testdata",