Searched refs:ShapeComputeModule (Results 1 – 1 of 1) sorted by relevance
179 class ShapeComputeModule(torch.nn.Module): class186 shape_compute_module = torch.jit.script(ShapeComputeModule())