提交 7750cc75 编写于 作者: D dengkaipeng

fix op_flops not define. test=develop

上级 f4eda869
...@@ -176,6 +176,7 @@ def count_element_op(op): ...@@ -176,6 +176,7 @@ def count_element_op(op):
def _graph_flops(graph, detail=False): def _graph_flops(graph, detail=False):
assert isinstance(graph, GraphWrapper) assert isinstance(graph, GraphWrapper)
flops = 0 flops = 0
op_flops = 0
table = Table(["OP Type", 'Param name', "Flops"]) table = Table(["OP Type", 'Param name', "Flops"])
for op in graph.ops(): for op in graph.ops():
param_name = '' param_name = ''
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册