init() — pytorch Function Reference

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

Function python CoreTensor Registration calls 1 called by 1

Entity Profile

CoreTensor→ Registration→ __init__() — pytorch Function Reference

Dependency Diagram

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

Relationship Graph

Source Code

benchmarks/distributed/ddp/benchmark.py lines 176–182

    def __init__(self, device, distributed_backend, bucket_size, model):
        super().__init__(
            device,
            distributed_backend,
            bucket_size,
        )
        self.model = model