Searched refs:SumOverBatchSize (Results 1 – 2 of 2) sorted by relevance
2635 class SumOverBatchSize(Reduce): class2651 super(SumOverBatchSize, self).__init__(2657 class SumOverBatchSizeMetricWrapper(SumOverBatchSize):
1947 metrics_module.SumOverBatchSize() if hasattr(loss_fn, 'reduction')