Searched refs:Float8Module (Results 1 – 1 of 1) sorted by relevance
589 class Float8Module(torch.nn.Module): class598 _ = torch.onnx.dynamo_export(Float8Module(), torch.randn(1, 2, 3, 4))