experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix nn_batchnorm_test test. Remove squeeze.

Browse Source 
PiperOrigin-RevId: 261386373
This commit is contained in:
Yifei Feng 2019-08-02 14:00:50 -07:00 committed by TensorFlower Gardener
parent 872cdf24cb
commit 50396e0d8e
1 changed files with 1 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

6
tensorflow/python/ops/nn_batchnorm_test.py
View File

@ -364,11 +364,7 @@ class SufficientStatisticsTest(test.TestCase):
if d in set(axes): if d in set(axes):
count *= x.shape[d] count *= x.shape[d]
if not keep_dims: if not keep_dims:
# Starting from numpy 1.17.0, squeeze no longer take None input with not shift = np.asarray(shift)
# None axis.
if not shift:
axis = None
shift = np.squeeze(shift, axis=axis)
return count, m_ss, v_ss, shift return count, m_ss, v_ss, shift
def _opSuffStats(self, x, axes, shift, keep_dims): def _opSuffStats(self, x, axes, shift, keep_dims):