Multi-GPU TFF simulation errors “Detected dataset reduce op in multi-GPU TFF simulation”

Question

I ran my code for an emotion detection model using Tensorflow Federated simulation. My code work perfectly fine using CPUs only. However, I received this error when trying t