未验证 提交 f2612462 编写于 作者: K Kaipeng Deng 提交者: GitHub

fix op_flops not define. test=develop (#36489)

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