message: Update use of squeeze_dims to axis in call to squeeze
  This was causing a warning every time a user created a MonitoredSession.
diff: @@ -1232,7 +1232,7 @@ def boolean_mask(tensor, mask, name="boolean_mask", axis=None):
  def _apply_mask_1d(reshaped_tensor, mask, axis=None):
  """Mask tensor along dimension 0 with a 1-D mask."""
  - indices = squeeze(where(mask), squeeze_dims=[1])
  + indices = squeeze(where(mask), axis=[1])
  return gather(reshaped_tensor, indices, axis=axis)
  with ops.name_scope(name, values=[tensor, mask]):
  