allequal() — pytorch Function Reference

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

Function python CoreTensor Registration called by 1

Entity Profile

CoreTensor→ Registration→ allequal() — pytorch Function Reference

Dependency Diagram

graph TD
  ea399e06_2fd2_1127_c1f1_5c4919c59dc9["allequal()"]
  32bc5158_e977_7f97_b62d_d5c3b8b333f7["main()"]
  32bc5158_e977_7f97_b62d_d5c3b8b333f7 -->|calls| ea399e06_2fd2_1127_c1f1_5c4919c59dc9
  style ea399e06_2fd2_1127_c1f1_5c4919c59dc9 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/distributed/ddp/benchmark.py lines 44–50

def allequal(iterator):
    iterator = iter(iterator)
    try:
        first = next(iterator)
    except StopIteration:
        return True
    return all(first == rest for rest in iterator)