提交 0ec6362a authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Fix typo in infer_shape methods.

上级 62a7e19a
......@@ -1126,7 +1126,7 @@ class Gemm(GemmRelated):
zout[0] = z
def infer_shape(self, node, input_shapes):
return [inputs_shapes[0]]
return [input_shapes[0]]
setup_z_Nz_Sz_inplace = """
if (%(_zout)s != %(_z)s)
......@@ -1752,7 +1752,7 @@ class Dot22(GemmRelated):
raise
def infer_shape(self, node, input_shapes):
return [[inputs_shapes[0][0], inputs_shapes[1][1]]]
return [[input_shapes[0][0], input_shapes[1][1]]]
setup_z_Nz_Sz = """
if ((NULL == %(_zout)s)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论