@staticmethod
def _param_shapes(sample_shape):
    return {'rate': ops.convert_to_tensor(sample_shape, dtype=dtypes.int32)}
@staticmethod
def _param_shapes(sample_shape):
    return {'rate': ops.convert_to_tensor(sample_shape, dtype=dtypes.int32)}