inductor_aten_bmm() — pytorch Function Reference
Architecture documentation for the inductor_aten_bmm() function in inductor_bmm.py from the pytorch codebase.
Entity Profile
Dependency Diagram
graph TD f89158cb_fcc9_8241_4e22_048353b4ffc9["inductor_aten_bmm()"] 3c19a66e_66bb_a804_2d9f_5eed5907c0cc["test_total_time()"] 3c19a66e_66bb_a804_2d9f_5eed5907c0cc -->|calls| f89158cb_fcc9_8241_4e22_048353b4ffc9 style f89158cb_fcc9_8241_4e22_048353b4ffc9 fill:#6366f1,stroke:#818cf8,color:#fff
Relationship Graph
Source Code
benchmarks/dynamo/microbenchmarks/inductor_bmm.py lines 10–11
def inductor_aten_bmm(a, b):
return torch.bmm(a, b)
Domain
Subdomains
Called By
Source
Frequently Asked Questions
What does inductor_aten_bmm() do?
inductor_aten_bmm() is a function in the pytorch codebase.
What calls inductor_aten_bmm()?
inductor_aten_bmm() is called by 1 function(s): test_total_time.
Analyze Your Own Codebase
Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.
Try Supermodel Free