提交 fd05ca2d 编写于 作者: W wangzhuo325

call matmul instead of batchmatmul

上级 95c81242
...@@ -15,8 +15,10 @@ ...@@ -15,8 +15,10 @@
# limitations under the License. # limitations under the License.
"""matmul""" """matmul"""
from akg.ops.nn import batchmatmul from akg.ops.nn import matmul
def MatMul(x1, x2, transpose_a=False, transpose_b=False): def MatMul(x1, x2, out_dtype, transpose_a=False, transpose_b=False):
"""matmul""" """matmul"""
return batchmatmul.batchmatmul(x1, x2, transpose_a, transpose_b) return matmul.matmul(x=x1, y=x2, b=None, out_dtype=out_dtype,
left_format="zN", right_format="zN", out_format="zN",
transpose_x=transpose_a, transpose_y=transpose_b)
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册