Searched refs:MatMulCell (Results 1 – 5 of 5) sorted by relevance
58 class MatMulCell(nn.Cell): class76 self.cell1 = MatMulCell()77 self.cell2 = MatMulCell()78 self.cell3 = MatMulCell()79 self.cell4 = MatMulCell()
26 class MatMulCell(nn.Cell): class28 super(MatMulCell, self).__init__()54 self.matmul_cell = MatMulCell()
59 class MatMulCell(nn.Cell): class89 self.cell1 = MatMulCell(strategy1, strategy2)
59 class MatMulCell(nn.Cell): class80 cell = MatMulCell(strategy1, strategy2, param)
291 class MatMulCell(nn.Cell): class304 cell = MatMulCell()307 cell = MatMulCell()