Searched refs:SumOverBatchSizeMetricWrapper (Results 1 – 2 of 2) sorted by relevance
3248 class SumOverBatchSizeMetricWrapper(SumOverBatchSize): class3261 super(SumOverBatchSizeMetricWrapper, self).__init__(name=name, dtype=dtype)3273 return super(SumOverBatchSizeMetricWrapper, self).update_state(3280 base_config = super(SumOverBatchSizeMetricWrapper, self).get_config()
2048 sum_metric = metrics.SumOverBatchSizeMetricWrapper(metric_fn)