rand_int_tensor() — pytorch Function Reference

Architecture documentation for the rand_int_tensor() function in huggingface.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration called by 1

Entity Profile

DynamoBenchmarks→ Configuration→ rand_int_tensor() — pytorch Function Reference

Dependency Diagram

graph TD
  99522623_9bfd_bf9f_64dc_57a37f41f30a["rand_int_tensor()"]
  179ecd7f_27e8_48db_95f4_85c2cda45c18["generate_inputs_for_model()"]
  179ecd7f_27e8_48db_95f4_85c2cda45c18 -->|calls| 99522623_9bfd_bf9f_64dc_57a37f41f30a
  style 99522623_9bfd_bf9f_64dc_57a37f41f30a fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/huggingface.py lines 295–303

def rand_int_tensor(device, low, high, shape):
    return torch.randint(
        low,
        high,
        shape,
        device=device,
        dtype=torch.int64,
        requires_grad=False,
    )