Searched refs:shape_compute_module (Results 1 – 3 of 3) sorted by relevance
29 shape_compute_module: torch.nn.Module,38 self.shape_compute_module = shape_compute_module51 …self.out_templates = self.shape_compute_module.prepare(self.ser_model, args) # type: ignore[opera…107 shape_compute_module,118 shape_compute_module,186 shape_compute_module = torch.jit.script(ShapeComputeModule())190 shape_compute_module.define("".join(real_shape_compute_lines))193 shape_compute_module,
91 auto shape_compute_module = in preprocess() local94 shape_compute_module._save_for_mobile(ss); in preprocess()
120 auto shape_compute_module = _load_for_mobile(ss); in init() local122 shape_compute_module.run_method("prepare", ser_model, inputs) in init()