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

add tensor infer function

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