message: Make compute_output_shape public in masked core layers
diff: @@ -210,7 +210,7 @@ class _MaskedConv(base.Layer):
  return self.activation(outputs)
  return outputs
  - def _compute_output_shape(self, input_shape):
  + def compute_output_shape(self, input_shape):
  input_shape = tensor_shape.TensorShape(input_shape).as_list()
  if self.data_format == 'channels_last':
  space = input_shape[1:-1]
  @@ -467,7 +467,7 @@ class MaskedFullyConnected(base.Layer):
  return self.activation(outputs) # pylint: disable=not-callable
  return outputs
  - def _compute_output_shape(self, input_shape):
  + def compute_output_shape(self, input_shape):
  input_shape = tensor_shape.TensorShape(input_shape)
  input_shape = input_shape.with_rank_at_least(2)
  if input_shape[-1].value is None:
  