Home / Function/ mm() — pytorch Function Reference

mm() — pytorch Function Reference

Architecture documentation for the mm() function in bench_mm_fusion.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration called by 3

Entity Profile

Dependency Diagram

graph TD
  fe1cc36a_6b6c_badf_4954_01fbf4301526["mm()"]
  1710ce31_f7a2_1b15_ff67_002c8be881fd["mm_add()"]
  1710ce31_f7a2_1b15_ff67_002c8be881fd -->|calls| fe1cc36a_6b6c_badf_4954_01fbf4301526
  37a0f22d_678f_c676_7bb9_c7482c462ae6["mm_relu()"]
  37a0f22d_678f_c676_7bb9_c7482c462ae6 -->|calls| fe1cc36a_6b6c_badf_4954_01fbf4301526
  6f68b04e_8de5_dbe9_8c34_ecbacc0af2f5["mm_add_relu()"]
  6f68b04e_8de5_dbe9_8c34_ecbacc0af2f5 -->|calls| fe1cc36a_6b6c_badf_4954_01fbf4301526
  style fe1cc36a_6b6c_badf_4954_01fbf4301526 fill:#6366f1,stroke:#818cf8,color:#fff

Relationship Graph

Source Code

benchmarks/dynamo/microbenchmarks/bench_mm_fusion.py lines 23–25

    def mm(a, b, bias):
        y = torch.mm(a, b)
        return y

Domain

Subdomains

Called By

Source

Frequently Asked Questions

What does mm() do?
mm() is a function in the pytorch codebase.
What calls mm()?
mm() is called by 3 function(s): mm_add, mm_add_relu, mm_relu.

Analyze Your Own Codebase

Get architecture documentation, dependency graphs, and domain analysis for your codebase in minutes.

Try Supermodel Free