tfma.post_export_metrics.root_mean_squared_error

This is the function that the user calls.