diff --git a/llvm/projects/keras/frontend/approxhpvm_translator.py b/llvm/projects/keras/frontend/approxhpvm_translator.py index 2cb8c5d74143e7c7ed6e9f536cb90f8ee8a468ad..b70b90cfac983f969c1b57928f5bb18dadb76b38 100644 --- a/llvm/projects/keras/frontend/approxhpvm_translator.py +++ b/llvm/projects/keras/frontend/approxhpvm_translator.py @@ -116,10 +116,9 @@ class DFG: self.last_node = cur_node - # Print the DFG in reverse postorder - def printDFG(self): - - print ("\n\n ****** Printing DFG ******* \n\n") + #Build and Print the DFG in reverse postorder + def buildDFG(self): + print ("\n\n ****** Traversing and Printing DFG ******* \n\n") visited_nodes = {} # Starting traversal at the DFG root node self.traverseNode(self.root_node, visited_nodes) @@ -855,8 +854,8 @@ def translate_to_approxhpvm(model, weights_dir, test_data=None, test_labels=None # NOTE: Add DNN layer to graph dfg.add_to_graph(layer) - # Print DFG in reverse postorder - dfg.printDFG() + # Build and Print DFG in reverse postorder + dfg.buildDFG() print ("test_data.shape = ", test_data.shape, "\n")