Searched refs:metric_value_fn (Results 1 – 1 of 1) sorted by relevance
280 def _aggregate_across_replicas(metrics_collections, metric_value_fn, *args): argument299 metric_value = metric_value_fn(distribution, *a)302 metric_value = metric_value_fn(distribution, *a)306 metric_value = metric_value_fn(distribution, *a)