docstr: Returns tensorflow data generator.
code: def get_tf_generator(self):
      """Returns tensorflow data generator."""
      return tf.data.Dataset.from_generator(
          self.data_generator,
          output_types=tf.dtypes.int32,
          output_shapes=(None, 3),
      ).prefetch(0)
  
imports: import abc
  import glob
  import logging
  import os
  import shelve
  import shutil
  import tempfile
  from datetime import datetime
  import numpy
  import tensorflow
  from .graph_partitioner import PARTITION_ALGO_REGISTRY, AbstractGraphPartitioner
label_desc: Use image node_id 10 for calling the PartitionDataManager obj's underlying member method code with example usage: obj.get_tf_generator() and returns: tf