Home / Function/ load() — pytorch Function Reference

load() — pytorch Function Reference

Architecture documentation for the load() function in common.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration calls 2

Entity Profile

Dependency Diagram

graph TD
  6390669e_1575_3d66_5d4e_f02d7218bd4b["load()"]
  1eff8423_1f23_d138_6815_07d8dc29a749["_normalize_bench_inputs()"]
  6390669e_1575_3d66_5d4e_f02d7218bd4b -->|calls| 1eff8423_1f23_d138_6815_07d8dc29a749
  3df8de63_d71a_714c_93a1_1a0e24c0f362["load()"]
  6390669e_1575_3d66_5d4e_f02d7218bd4b -->|calls| 3df8de63_d71a_714c_93a1_1a0e24c0f362
  style 6390669e_1575_3d66_5d4e_f02d7218bd4b fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 1473–1488

    def load(cls, model, example_inputs):
        key = weakref.ref(model)
        if key not in cls.cache:
            example_args, example_kwargs = _normalize_bench_inputs(example_inputs)
            if example_args:
                jit_traced_module = torch.jit.trace(
                    model, example_inputs=example_args, strict=False
                )
            else:
                jit_traced_module = torch.jit.trace(
                    model, example_kwarg_inputs=example_kwargs, strict=False
                )

            cls.cache[key] = jit_traced_module

        return cls.cache[key]

Domain

Subdomains

Calls

Source

Frequently Asked Questions

What does load() do?
load() is a function in the pytorch codebase.
What does load() call?
load() calls 2 function(s): _normalize_bench_inputs, load.

Analyze Your Own Codebase

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

Try Supermodel Free