提交 0653b926 authored 作者: Melanie Ducoffe's avatar Melanie Ducoffe

put Alloc empty to init zero in local_dot22_to_ger_or_gemv

上级 94732dcd
...@@ -1818,7 +1818,8 @@ def local_dot22_to_ger_or_gemv(node): ...@@ -1818,7 +1818,8 @@ def local_dot22_to_ger_or_gemv(node):
xb = x.broadcastable xb = x.broadcastable
yb = y.broadcastable yb = y.broadcastable
one = T.as_tensor_variable(numpy.asarray(1, dtype=x.dtype)) one = T.as_tensor_variable(numpy.asarray(1, dtype=x.dtype))
zero = T.as_tensor_variable(numpy.asarray(0, dtype=x.dtype)) # zero = T.as_tensor_variable(numpy.asarray(0, dtype=x.dtype))
zero = T.AllocEmpty(x.dtype)(1)
if xb[1] and yb[0]: if xb[1] and yb[0]:
# x and y are both vectors so this might qualifies for a GER # x and y are both vectors so this might qualifies for a GER
xv = x.dimshuffle(0) xv = x.dimshuffle(0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论