Searched refs:SumOverBatchSizeMetricWrapper (Results 1 – 2 of 2) sorted by relevance
2657 class SumOverBatchSizeMetricWrapper(SumOverBatchSize): class2670 super(SumOverBatchSizeMetricWrapper, self).__init__(name=name, dtype=dtype)2681 return super(SumOverBatchSizeMetricWrapper, self).update_state(2688 base_config = super(SumOverBatchSizeMetricWrapper, self).get_config()
1948 else metrics_module.SumOverBatchSizeMetricWrapper(loss_fn)