Searched refs:LayerNormModule (Results 1 – 1 of 1) sorted by relevance
14 class LayerNormModule(torch.nn.Module): class362 "norm": lambda: (LayerNormModule(), LayerNormModule.get_example_inputs()),