Question: what does tf group do?
Answer: group arguments belong to multiple devices, tf. group will insert an intermediate layer between its inputs and the node it returns. That layer will contain one node per device, so that the dependencies are organized by device. This could reduce the cross-device data flow.