Replace deprecated RemoveDuplicates API with Distinct.
Replace deprecated RemoveDuplicates API with Distinct.

PiperOrigin-RevId: 254214330

tensorflow/model-analysis,tensorflow/model-analysis,tensorflow/model-analysis,tensorflow/model-analysis