From c9862c3e1e507b0517dcd432dc674b01911e7f99 Mon Sep 17 00:00:00 2001 From: "A. Unique TensorFlower" Date: Mon, 12 Nov 2018 22:18:06 -0800 Subject: [PATCH] Get "distribute" library files out of the "training" target. PiperOrigin-RevId: 221212516 --- tensorflow/python/BUILD | 3 +++ 1 file changed, 3 insertions(+) diff --git a/tensorflow/python/BUILD b/tensorflow/python/BUILD index aa78fc157a5..4fe92262ba6 100644 --- a/tensorflow/python/BUILD +++ b/tensorflow/python/BUILD @@ -3423,6 +3423,8 @@ py_library( # file): "training/basic_session_run_hooks.py", "training/checkpoint_management.py", + "training/distribute.py", + "training/distribution_strategy_context.py", "training/saveable_object.py", "training/saver.py", "training/session_run_hook.py", @@ -3440,6 +3442,7 @@ py_library( ":control_flow_ops", ":data_flow_ops", ":device", + ":distribute", ":errors", ":framework", ":framework_for_generated_wrappers",