tfma.metrics.merge_per_key_computations

View source on GitHub

Wraps create_computations_fn to be called separately for each key.