def _population_count(x):
    orig_dtype = x.dtype
    return tf.cast(tf.raw_ops.PopulationCount(x=x), orig_dtype)
def _population_count(x):
    orig_dtype = x.dtype
    return tf.cast(tf.raw_ops.PopulationCount(x=x), orig_dtype)