Home / Function/ print_aten_ops() — pytorch Function Reference

print_aten_ops() — pytorch Function Reference

Architecture documentation for the print_aten_ops() function in common.py from the pytorch codebase.

Function python DynamoBenchmarks Microbenchmarking

Entity Profile

Relationship Graph

Source Code

benchmarks/dynamo/common.py lines 1235–1242

def print_aten_ops(gm, example_inputs):
    from functorch.compile import aot_module

    def trace_printer(gm, _):
        print(gm.graph)
        return gm

    return aot_module(gm, fw_compiler=trace_printer, bw_compiler=trace_printer)

Domain

Subdomains

Source

Analyze Your Own Codebase

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

Try Supermodel Free