Searched refs:BatchMatMulNet (Results 1 – 2 of 2) sorted by relevance
26 class BatchMatMulNet(nn.Cell): class28 super(BatchMatMulNet, self).__init__()49 net = BatchMatMulNet()70 net = BatchMatMulNet()95 net = BatchMatMulNet(transpose_a=True)120 net = BatchMatMulNet(transpose_b=True)145 net = BatchMatMulNet(transpose_a=True, transpose_b=True)
27 class BatchMatMulNet(nn.Cell): class29 super(BatchMatMulNet, self).__init__()43 net = BatchMatMulNet()65 net = BatchMatMulNet()87 net = BatchMatMulNet(transpose_a=True)109 net = BatchMatMulNet(transpose_b=True)131 net = BatchMatMulNet(transpose_a=True, transpose_b=True)153 net = BatchMatMulNet()