@ -71,5 +71,5 @@ from mindspore.ops.op_info_register import op_info_register
]
}""")
def CusBatchMatMul(input_x1, input_x2, output, transpose_a=False, transpose_b=True, kernel_name="batchmatmul"):
"""CusBatchMatMul"""
return
@ -59,5 +59,5 @@ from mindspore.ops.op_info_register import op_info_register
def CusCholeskyTrsm(input_x, output, kernel_name):
"""CusCholeskyTrsm"""
@ -35,4 +35,3 @@ class CusBatchMatMul(PrimitiveWithInfer):
def infer_dtype(self, data1_dtype, data2_dtype):
return data1_dtype
@ -35,4 +35,3 @@ class CusCholeskyTrsm(PrimitiveWithInfer):
def infer_dtype(self, data1_dtype):
@ -42,4 +42,3 @@ class CusFusedAbsMax1(PrimitiveWithInfer):
@ -49,4 +49,3 @@ class CusImg2Col(PrimitiveWithInfer):
@ -52,4 +52,3 @@ class CusMatMulCube(PrimitiveWithInfer):
return ms.common.dtype.tensor_type(getattr(ms, "float32"))