message: Fix deprecation warning when calling boolean_mask
diff: @@ -1472,7 +1472,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), axis=[1])
  + indices = squeeze(where_v2(mask), axis=[1])
  return gather(reshaped_tensor, indices, axis=axis)
  with ops.name_scope(name, values=[tensor, mask]):
  