Add ability to specify num_steps_per_batch to generate accurate number of examples/sec.
Add ability to specify num_steps_per_batch to generate accurate number of examples/sec.

PiperOrigin-RevId: 283623968
Change-Id: Icadf52490aa6494bfa9d94aea0cc4af2db110707

tensorflow/agents,tensorflow/agents