aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/__init__.py
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2018-03-29 15:28:24 -0700
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2018-03-29 15:32:01 -0700
commit6f5d7a97cd2c0741ddfa756853ce5321377b5d53 (patch)
treee79afd91cd68bc9ed75bfe278511312da3918fe6 /tensorflow/contrib/__init__.py
parent40f8291db5c0b05b31d7bbe23b847cdbb2408718 (diff)
Add tf.contrib.distribute, which defines classes DistributionStrategy
and MirroredStrategy, and related functionality. Also add tf.contrib.optimizer_v2, an update to the Optimizer API. RELNOTES: Can now pass tf.contrib.distribute.MirroredStrategy() to tf.estimator.RunConfig() to run an Estimator model on multiple GPUs on one machine. PiperOrigin-RevId: 190996247
Diffstat (limited to 'tensorflow/contrib/__init__.py')
-rw-r--r--tensorflow/contrib/__init__.py2
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/contrib/__init__.py b/tensorflow/contrib/__init__.py
index 4f6f539027..a8e05df708 100644
--- a/tensorflow/contrib/__init__.py
+++ b/tensorflow/contrib/__init__.py
@@ -30,6 +30,7 @@ from tensorflow.contrib import crf
from tensorflow.contrib import cudnn_rnn
from tensorflow.contrib import data
from tensorflow.contrib import deprecated
+from tensorflow.contrib import distribute
from tensorflow.contrib import distributions
from tensorflow.contrib import estimator
from tensorflow.contrib import factorization
@@ -84,6 +85,7 @@ from tensorflow.contrib import training
from tensorflow.contrib import util
from tensorflow.contrib.eager.python import tfe as eager
from tensorflow.contrib.lite.python import lite
+from tensorflow.contrib.optimizer_v2 import optimizer_v2_symbols as optimizer_v2
from tensorflow.contrib.receptive_field import receptive_field_api as receptive_field
from tensorflow.contrib.remote_fused_graph import pylib as remote_fused_graph
from tensorflow.contrib.specs import python as specs