Add option to disable use of tf.function in metric_utils.eager_compute.
Add option to disable use of tf.function in metric_utils.eager_compute.

PiperOrigin-RevId: 271380205
Change-Id: Id7c9732363d4d4d7583fdfabb524d8cd4f661689

tensorflow/agents,tensorflow/agents