Brooks-type theorems for relaxations of square colorings