aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/distributions/__init__.py
diff options
context:
space:
mode:
Diffstat (limited to 'tensorflow/contrib/distributions/__init__.py')
-rw-r--r--tensorflow/contrib/distributions/__init__.py13
1 files changed, 5 insertions, 8 deletions
diff --git a/tensorflow/contrib/distributions/__init__.py b/tensorflow/contrib/distributions/__init__.py
index 257aefa857..470b9edb79 100644
--- a/tensorflow/contrib/distributions/__init__.py
+++ b/tensorflow/contrib/distributions/__init__.py
@@ -90,9 +90,10 @@ from __future__ import print_function
# pylint: disable=unused-import,wildcard-import,line-too-long,g-importing-member
-from tensorflow.contrib.distributions.python.ops import bijectors
+
from tensorflow.contrib.distributions.python.ops.bernoulli import *
from tensorflow.contrib.distributions.python.ops.beta import *
+from tensorflow.contrib.distributions.python.ops.bijectors import *
from tensorflow.contrib.distributions.python.ops.binomial import *
from tensorflow.contrib.distributions.python.ops.categorical import *
from tensorflow.contrib.distributions.python.ops.chi2 import *
@@ -134,12 +135,8 @@ from tensorflow.contrib.distributions.python.ops.wishart import *
from tensorflow.python.util.all_util import remove_undocumented
-_allowed_symbols = [
- 'bijectors',
- 'ConditionalDistribution',
- 'ConditionalTransformedDistribution',
- 'FULLY_REPARAMETERIZED',
- 'NOT_REPARAMETERIZED',
-]
+_allowed_symbols = ['ConditionalDistribution',
+ 'ConditionalTransformedDistribution',
+ 'FULLY_REPARAMETERIZED', 'NOT_REPARAMETERIZED']
remove_undocumented(__name__, _allowed_symbols)