提交 ddfb387c authored 作者: Frederic's avatar Frederic

Print the optimizer profile on the right destination.

上级 e02ce0d8
......@@ -787,8 +787,8 @@ class ProfileStats(object):
if self.variable_shape or self.variable_strides:
self.summary_memory(file, n_apply_to_print)
if self.optimizer_profile:
print "Optimizer Profile"
print "-----------------"
print >> file, "Optimizer Profile"
print >> file, "-----------------"
self.optimizer_profile[0].print_profile(file,
self.optimizer_profile[1])
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论