提交 f99578af authored 作者: Frederic Bastien's avatar Frederic Bastien

Clean up and small change found while doing the previous commit

上级 a867f1f3
...@@ -311,7 +311,9 @@ class SeqOptimizer(Optimizer, list): ...@@ -311,7 +311,9 @@ class SeqOptimizer(Optimizer, list):
print(blanc, " ", i[0], ',', i[1], file=stream) print(blanc, " ", i[0], ',', i[1], file=stream)
if level == 0: if level == 0:
print(blanc, " time - (name, class, index, nodes before, nodes after) - validate time", file=stream) print(blanc,
" time - (name, class, index, nodes before, nodes after) - validate time",
file=stream)
ll = [] ll = []
for opt in opts: for opt in opts:
if hasattr(opt, "__name__"): if hasattr(opt, "__name__"):
...@@ -864,7 +866,9 @@ class MergeOptimizer(Optimizer): ...@@ -864,7 +866,9 @@ class MergeOptimizer(Optimizer):
callbacks_time = {} callbacks_time = {}
for k, v in iteritems(fgraph.execute_callbacks_times): for k, v in iteritems(fgraph.execute_callbacks_times):
if k in callbacks_before: if k in callbacks_before:
callbacks_time[k] = v - callbacks_before[k] t = v - callbacks_before[k]
if t > 0:
callbacks_time[k] = t
else: else:
callbacks_time[k] = v callbacks_time[k] = v
else: else:
...@@ -894,7 +898,9 @@ class MergeOptimizer(Optimizer): ...@@ -894,7 +898,9 @@ class MergeOptimizer(Optimizer):
print(blanc, " callbacks_time", file=stream) print(blanc, " callbacks_time", file=stream)
for i in sorted(iteritems(callbacks_time), key=lambda a: a[1]): for i in sorted(iteritems(callbacks_time), key=lambda a: a[1]):
if i[1] > 0: if i[1] > 0:
print(i) # We want to have the __str__ called, so we can't
# just print i.
print(blanc, " ", i[0], ',', i[1], file=stream)
@staticmethod @staticmethod
def merge_profile(prof1, prof2): def merge_profile(prof1, prof2):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论