diff options
author | Dustin Tran <trandustin@google.com> | 2018-01-19 12:00:33 -0800 |
---|---|---|
committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-01-19 12:06:39 -0800 |
commit | 38a1921ae911d2a7ade487bf3f80ba932dd53988 (patch) | |
tree | 16afa09f965352022c403365b330f0fb3fd31805 /tensorflow/contrib/bayesflow | |
parent | 4cf9eee73beb5bf5105047d8be3832a064107324 (diff) |
Move layers_dense_variational_impl.py to layers_dense_variational.py.
PiperOrigin-RevId: 182568201
Diffstat (limited to 'tensorflow/contrib/bayesflow')
-rw-r--r-- | tensorflow/contrib/bayesflow/python/kernel_tests/layers_dense_variational_test.py | 2 | ||||
-rw-r--r-- | tensorflow/contrib/bayesflow/python/ops/layers.py | 2 | ||||
-rw-r--r-- | tensorflow/contrib/bayesflow/python/ops/layers_dense_variational.py (renamed from tensorflow/contrib/bayesflow/python/ops/layers_dense_variational_impl.py) | 17 |
3 files changed, 2 insertions, 19 deletions
diff --git a/tensorflow/contrib/bayesflow/python/kernel_tests/layers_dense_variational_test.py b/tensorflow/contrib/bayesflow/python/kernel_tests/layers_dense_variational_test.py index 4e9f119351..342f38ccec 100644 --- a/tensorflow/contrib/bayesflow/python/kernel_tests/layers_dense_variational_test.py +++ b/tensorflow/contrib/bayesflow/python/kernel_tests/layers_dense_variational_test.py @@ -20,7 +20,7 @@ from __future__ import print_function import numpy as np -from tensorflow.contrib.bayesflow.python.ops import layers_dense_variational_impl as prob_layers_lib +from tensorflow.contrib.bayesflow.python.ops import layers_dense_variational as prob_layers_lib from tensorflow.contrib.bayesflow.python.ops import layers_util as prob_layers_util from tensorflow.contrib.distributions.python.ops import independent as independent_lib from tensorflow.python.framework import dtypes diff --git a/tensorflow/contrib/bayesflow/python/ops/layers.py b/tensorflow/contrib/bayesflow/python/ops/layers.py index 1bc1af8659..a742b7c1aa 100644 --- a/tensorflow/contrib/bayesflow/python/ops/layers.py +++ b/tensorflow/contrib/bayesflow/python/ops/layers.py @@ -24,7 +24,7 @@ from __future__ import print_function # go/tf-wildcard-import # pylint: disable=wildcard-import from tensorflow.contrib.bayesflow.python.ops.layers_conv_variational import * -from tensorflow.contrib.bayesflow.python.ops.layers_dense_variational_impl import * +from tensorflow.contrib.bayesflow.python.ops.layers_dense_variational import * from tensorflow.contrib.bayesflow.python.ops.layers_util import * # pylint: enable=wildcard-import from tensorflow.python.util.all_util import remove_undocumented diff --git a/tensorflow/contrib/bayesflow/python/ops/layers_dense_variational_impl.py b/tensorflow/contrib/bayesflow/python/ops/layers_dense_variational.py index ec6e6ff977..591a8e553d 100644 --- a/tensorflow/contrib/bayesflow/python/ops/layers_dense_variational_impl.py +++ b/tensorflow/contrib/bayesflow/python/ops/layers_dense_variational.py @@ -13,13 +13,6 @@ # limitations under the License. # ============================================================================== """Dense Bayesian layer using KL-divergence based variational inference. - -@@DenseReparameterization -@@DenseLocalReparameterization -@@DenseFlipout -@@dense_reparameterization -@@dense_local_reparameterization -@@dense_flipout """ from __future__ import absolute_import @@ -40,16 +33,6 @@ from tensorflow.python.ops.distributions import normal as normal_lib from tensorflow.python.ops.distributions import util as distribution_util -__all__ = [ - "DenseReparameterization", - "DenseLocalReparameterization", - "DenseFlipout", - "dense_reparameterization", - "dense_local_reparameterization", - "dense_flipout", -] - - class _DenseVariational(layers_lib.Layer): """Abstract densely-connected class (private, used as implementation base). |