Searched refs:AddReluMod (Results 1 – 1 of 1) sorted by relevance
627 class AddReluMod(torch.nn.Module): class635 for name, mod in [("add", AddMod), ("add_relu", AddReluMod), ("mul", MulMod)]: