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

in pydotprint, go take the mode by default in the function automatically.

上级 821f69e5
...@@ -400,6 +400,8 @@ def pydotprint(fct, outfile=None, ...@@ -400,6 +400,8 @@ def pydotprint(fct, outfile=None,
if outfile is None: if outfile is None:
outfile = os.path.join(config.compiledir,'theano.pydotprint.' + outfile = os.path.join(config.compiledir,'theano.pydotprint.' +
config.device + '.' + format) config.device + '.' + format)
if mode is None:
mode = fct.maker.mode
if not isinstance(mode,ProfileMode) or not mode.fct_call.has_key(fct): if not isinstance(mode,ProfileMode) or not mode.fct_call.has_key(fct):
mode=None mode=None
try: try:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论