提交 8c84cd16 authored 作者: Razvan Pascanu's avatar Razvan Pascanu

make use of the profile string if one provided

上级 e1edbefd
...@@ -357,7 +357,10 @@ class Scan(PureOp): ...@@ -357,7 +357,10 @@ class Scan(PureOp):
profile = None profile = None
if (theano.config.profile or (isinstance(self.profile, (str,bool)) if (theano.config.profile or (isinstance(self.profile, (str,bool))
and self.profile)): and self.profile)):
profile = ScanProfileStats(name = self.name) if isinstance(self.profile, str):
profile = ScanProfileStats(name = self.profile)
else:
profile = ScanProfileStats(name = self.name)
elif self.profile: elif self.profile:
profile = self.profile profile = self.profile
self.fn = function(wrapped_inputs, self.fn = function(wrapped_inputs,
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论