Searched refs:traced_script_module (Results 1 – 5 of 5) sorted by relevance
13 traced_script_module = torch.jit.trace(model, example)14 optimized_module = optimize_for_mobile(traced_script_module)30 traced_script_module = torch.jit.trace(model, example)31 optimized_module = optimize_for_mobile(traced_script_module, backend="vulkan")47 traced_script_module = torch.jit.trace(model, example)48 optimized_module = optimize_for_mobile(traced_script_module)
18 traced_script_module = torch.jit.trace(model, example) variable21 traced_script_module.save("MobileNetV2.pt")24 ops = torch.jit.export_opnames(traced_script_module)
17 traced_script_module = torch.jit.trace(model, example) variable20 traced_script_module.save("MobileNetV2.pt")23 ops = torch.jit.export_opnames(traced_script_module)
10 traced_script_module = torch.jit.trace(model, example) variable11 ops = torch.jit.export_opnames(traced_script_module)
10 traced_script_module = torch.jit.trace(model, example) variable11 optimized_scripted_module = optimize_for_mobile(traced_script_module)