[Fix] set the option `ret_grouped_xyz`

[jeasinema]

The GroupAll module receives ret_grouped_xyz in its constructor but never sets it as a member. However, self.ret_grouped_xyz is still being used in the forward call. This PR fixes this issue.