forked from mindspore-Ecosystem/mindspore
delete bias from matmul
This commit is contained in:
parent
8346da267b
commit
960f5c5250
|
@ -648,7 +648,7 @@ class MatMul(PrimitiveWithInfer):
|
|||
raise ValueError('MatMul input x, y should be the same dimension size and should be '
|
||||
+ f'equal to 2, while x size = {len(x)}, y size= {len(y)}')
|
||||
|
||||
def infer_shape(self, x, y, bias=None):
|
||||
def infer_shape(self, x, y):
|
||||
self.check_shape_size(x, y)
|
||||
cls_name = self.name
|
||||
# expected dimension of x, y, x:[...,a,b] y:[..., c,d], the dim size should be the same except the last two
|
||||
|
@ -673,7 +673,7 @@ class MatMul(PrimitiveWithInfer):
|
|||
ret_dims = x[: -2] + [x_last[self.transpose_a], y_last[not self.transpose_b]]
|
||||
return ret_dims
|
||||
|
||||
def infer_dtype(self, x, y, bias=None):
|
||||
def infer_dtype(self, x, y):
|
||||
args = {"x": x, "y": y}
|
||||
validator.check_tensor_type_same(args, mstype.float_type + mstype.int_type, self.name)
|
||||
if x.element_type() == mstype.int8:
|
||||
|
|
Loading…
Reference in New Issue