提交 29a13abc 编写于 作者: J jiangjiajun

add tensor infer function

上级 458cc1dc
...@@ -177,7 +177,7 @@ class PaddleEmitter(object): ...@@ -177,7 +177,7 @@ class PaddleEmitter(object):
code.append("import paddle.fluid as fluid") code.append("import paddle.fluid as fluid")
code.append("import numpy") code.append("import numpy")
code.append("") code.append("")
code.append("def KitModel():") code.append("def Model():")
return code return code
def add_codes(self, indent, codes): def add_codes(self, indent, codes):
...@@ -233,7 +233,6 @@ class PaddleEmitter(object): ...@@ -233,7 +233,6 @@ class PaddleEmitter(object):
# merge all the generated python codes # merge all the generated python codes
for node in translate_nodes: for node in translate_nodes:
print(self.graph.get_node(node).output_name, node)
codes = self.graph.get_node(node).code.gen_codes() codes = self.graph.get_node(node).code.gen_codes()
self.add_codes(1, codes) self.add_codes(1, codes)
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册