[tf.data] Calculate the average input time for the root node of the data input pipeline. Also add a unit test for the `SelfProcessingTime` function.

PiperOrigin-RevId: 322686110
Change-Id: I72c0bddb2ed32570325c7d8023aaaffbeab87378