提交 274cf81d authored 作者: Anish Shah's avatar Anish Shah

tip for mixed dtypes

上级 8a736df5
......@@ -1391,6 +1391,16 @@ class ProfileStats(object):
" generator supported on the GPU.", file = file)
break
# tip 6
for a in self.apply_time:
node = a
if (isinstance(node.op, T.Dot) and
len(set(i.dtype for i in node.inputs)) != 1):
print(" - You have a dot operation that has different dtype "
" for inputs (%s). Make sure that the inputs have same "
" dtype." % [i.type for i in node.inputs], file = file)
printed_tip = True
if not printed_tip:
print(" Sorry, no tip for today.", file = file)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论