keras/src/metrics/iou_metrics.py/_IoUBase/reset_state
class _IoUBase:    def reset_state(self):
        self.total_cm.assign(
            ops.zeros(self.total_cm.shape, dtype=self.total_cm.dtype)
        )