def METHOD_NAME(self, batch_shape):
    try:
        return super().METHOD_NAME(batch_shape)
    except NotImplementedError:
        validate_args = self.__dict__.get("_validate_args")
        loc = self.loc.METHOD_NAME(batch_shape)
        concentration = self.concentration.METHOD_NAME(batch_shape)
        return type(self)(loc, concentration, validate_args=validate_args)