Partitioning Functional API Call in TensorFlow

Graph partitioning has the following general flow:
  1. Create/initialize the partition class:
    from vai.dpuv1.rt.xdnn_rt_tf import TFxdnnRT
xdnnTF = TFxdnnRT(args)
  2. Loading the partitioned graph:
    graph = xdnnTF.load_partitioned_graph()
  3. Apply preprocessing and post processing as if the original graph is loaded.