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

fix the upcast to use the good inputs.

上级 1d468fa5
......@@ -806,9 +806,8 @@ class SamplingDotCsr(gof.Op):
raise NotImplementedError(
'Complex types are not supported for pattern')
# TODO: why 2 times the same inputs?
dot_out = scalar.upcast(node.inputs[0].type.dtype,
node.inputs[0].type.dtype)
node.inputs[1].type.dtype)
if dot_out == "float32":
conv_type = "float"
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论