Skip the trajectory_collection/misc time, this is just confusing.
Skip the trajectory_collection/misc time, this is just confusing.

Right now we do:
                    trajectory_collection :     877.29
        trajectory_collection/env_actions :     261.57
               trajectory_collection/misc :     682.60
 trajectory_collection/policy_application :     193.81

And ideally we'd like sum of trajectory_collection/* to be <= the value of trajectory_collection, with the rest as misc.

PiperOrigin-RevId: 255711981

tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor,tensorflow/tensor2tensor