Searched refs:TrainOneStepWithLarsCell (Results 1 – 1 of 1) sorted by relevance
45 class TrainOneStepWithLarsCell(nn.Cell): class49 super(TrainOneStepWithLarsCell, self).__init__(auto_prefix=False)87 TrainOneStepWithLarsCell(_network, opt)