Searched refs:CustomMatMul (Results 1 – 1 of 1) sorted by relevance
52 class CustomMatMul(nn.Cell): class54 super(CustomMatMul, self).__init__()65 self.fc = CustomMatMul(transpose_b=True)66 self.fc1 = CustomMatMul(transpose_b=True)67 self.fc2 = CustomMatMul(transpose_b=True)68 self.fc3 = CustomMatMul(transpose_b=True)69 self.fc4 = CustomMatMul(transpose_b=True)