Searched refs:ShapeModule (Results 1 – 2 of 2) sorted by relevance
698 class ShapeModule(torch.nn.Module): class711 ShapeModule(), (x,), input_names=["x"], dynamic_axes={"x": [0, 1]}
9660 class ShapeModule(torch.nn.Module): class9670 self.run_test(ShapeModule(), (x,), rtol=1e-3, atol=1e-5)