get_shapes() — pytorch Function Reference
Architecture documentation for the get_shapes() function in utils.py from the pytorch codebase.
Entity Profile
Relationship Graph
Source Code
benchmarks/dynamo/genai_layers/utils.py lines 61–63
def get_shapes(self) -> tuple[tuple[int, ...], ...]:
# Get a list of input shapes to benchmark the kernel
raise NotImplementedError
Domain
Subdomains
Source
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free