#tf-data-service Merge duplicate method `GetIteratorContextWithIndexMapper`.
#tf-data-service Merge duplicate method `GetIteratorContextWithIndexMapper`.

Also add a simple class to select the right `IteratorContext` when index
shuffling is needed.
