message: Undo import removal
  Summary: D20224 removed the import.
  Test Plan: Tests pass
  Reviewers: #tensorflow, jakeh
diff: # pylint: disable=wildcard-import,unused-import
  from tensorflow.python.ipu import autoshard
  from tensorflow.python.ipu import autoshard_cnn
  +from tensorflow.python.ipu import data
  from tensorflow.python.ipu import dataset_benchmark
  from tensorflow.python.ipu import ipu_compiler
  from tensorflow.python.ipu import ipu_infeed_queue
  