Fix the order of applying sampling correction and temperature in the Retrieval task.
Fix the order of applying sampling correction and temperature in the Retrieval task.

PiperOrigin-RevId: 362091982

tensorflow/recommenders,tensorflow/recommenders