diff options
| author |  Priya Gupta <priyag@google.com> | 2018-08-14 11:22:19 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-08-14 11:31:36 -0700 |
| commit | 77fabbeabb5b9061d8c606050c1ea79aec990c03 (patch) | |
| tree | 1495d6acb396eebd40c703b891a4f2e7437a8532 /tensorflow/python/BUILD | |
| parent | cea262e16a004d73295259c42f21e2655da3df13 (diff) | |
1. Move distribution strategy context utility methods to a separate file with few dependencies. This allows us to import this in some places without creating circular dependencies as the original file imported many things.
2. Move the stack used in distribution strategy context to the graph. This allows us to use different strategies in different graphs (for e.g. in train and eval).
This fixes #21412 and #21180.
PiperOrigin-RevId: 208680454
Diffstat (limited to 'tensorflow/python/BUILD')
| -rw-r--r-- | tensorflow/python/BUILD | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index 2a71eaf030..a6bb6158e6 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3340,7 +3340,10 @@ py_library( py_library( name = "distribute", - srcs = ["training/distribute.py"], + srcs = [ + "training/distribute.py", + "training/distribution_strategy_context.py", + ], srcs_version = "PY2AND3", deps = [ ":array_ops", |