pytorch Architecture Docs
Home / Function/ scriptAndSave() — pytorch Function Reference

scriptAndSave() — pytorch Function Reference

Architecture documentation for the scriptAndSave() function in generate_test_torchscripts.py from the pytorch codebase.

Function python AndroidRuntime LifeCycleTesting

Entity Profile

Relationship Graph

Source Code

android/pytorch_android/generate_test_torchscripts.py lines 10–18

def scriptAndSave(module, fileName):
    print("-" * 80)
    script_module = torch.jit.script(module)
    print(script_module.graph)
    outputFileName = OUTPUT_DIR + fileName
    # note that the lite interpreter model can also be used in full JIT
    script_module._save_for_lite_interpreter(outputFileName)
    print("Saved to " + outputFileName)
    print("=" * 80)

Domain

Subdomains

Source

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free