提交 f99ca983 authored 作者: Tim Cooijmans's avatar Tim Cooijmans

use AllocEmpty for unused input

上级 0ec4a230
......@@ -746,7 +746,7 @@ def local_gpua_gemm(node, context_name):
@op_lifter([tensor.blas.BatchedDot])
def local_gpua_gemmbatch(node, context_name):
a, b = node.inputs
c = tensor.zeros((a.shape[0], a.shape[1], b.shape[2]))
c = tensor.AllocEmpty((a.shape[0], a.shape[1], b.shape[2]))
return gpugemmbatch_no_inplace(c, 1.0, a, b, 0.0)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论