init() — pytorch Function Reference

Architecture documentation for the __init__() function in dist_util.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking calls 1

Entity Profile

DynamoBenchmarks→ Microbenchmarking→ __init__() — pytorch Function Reference

Dependency Diagram

graph TD
  c5c85bd8_9287_390c_b15a_ddc19fe82dc8["__init__()"]
  d7622583_7435_0c2d_689b_eab6ddfd0498["__init__()"]
  c5c85bd8_9287_390c_b15a_ddc19fe82dc8 -->|calls| d7622583_7435_0c2d_689b_eab6ddfd0498
  style c5c85bd8_9287_390c_b15a_ddc19fe82dc8 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/dist_util.py lines 59–74

    def __init__(self):
        super().__init__()
        self.net = nn.Sequential(
            *[nn.Linear(10, 10000), nn.ReLU()]
            + [nn.Linear(10000, 10000), nn.ReLU()]
            + [MyModule(10000, 10000)]
            + [MyModule(10000, 1000)]
            + [MyModule(1000, 1000)]
            + [MyModule(1000, 1000)]
            + [MyModule(1000, 1000)]
            + [MyModule(1000, 1000)]
            + [MyModule(1000, 1000)]
            + [MyModule(1000, 1000)]
            + [MyModule(1000, 1000)]
            + [nn.Linear(1000, 5)]
        )