Searched refs:BatchMatMul_d (Results 1 – 1 of 1) sorted by relevance
167 class BatchMatMul_d(nn.Cell): class169 super(BatchMatMul_d, self).__init__()185 net = BatchMatMul_d()