Partitioning Functional API Call in TensorFlow - 1.4 English

Vitis AI User Guide (UG1414)

Document ID
UG1414
Release Date
2021-07-22
Version
1.4 English
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.