diff options
| author |  Patrick Nguyen <drpng@google.com> | 2018-05-01 14:28:36 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-05-01 14:33:20 -0700 |
| commit | 325d0ef21a48bea1cc618a2bd24a9776de417ce5 (patch) | |
| tree | d41cf6304071e95bebd5747ca87dfca571e98634 /tensorflow/contrib/factorization | |
| parent | 46bf1e8934b3bc8edeff3f218a50b0ee5806e96b (diff) | |
Merge changes from github.
PiperOrigin-RevId: 194997009
Diffstat (limited to 'tensorflow/contrib/factorization')
| -rw-r--r-- | tensorflow/contrib/factorization/python/ops/gmm_ops.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/factorization/python/ops/gmm_ops.py b/tensorflow/contrib/factorization/python/ops/gmm_ops.py index ccdd679d6a..e076631bc1 100644 --- a/tensorflow/contrib/factorization/python/ops/gmm_ops.py +++ b/tensorflow/contrib/factorization/python/ops/gmm_ops.py @@ -397,7 +397,7 @@ class GmmAlgorithm(object): # Compute the effective number of data points assigned to component k. with ops.control_dependencies(self._w): points_in_k = array_ops.squeeze( - math_ops.add_n(self._points_in_k), squeeze_dims=[0]) + math_ops.add_n(self._points_in_k), axis=[0]) # Update alpha. if 'w' in self._params: final_points_in_k = points_in_k / num_batches |