Home / Function/ inductor_aten_bmm() — pytorch Function Reference

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)

Subdomains

Called By

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