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