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