Home / Function/ eager() — pytorch Function Reference

eager() — pytorch Function Reference

Architecture documentation for the eager() function in kernels.py from the pytorch codebase.

Function python DynamoBenchmarks Configuration

Entity Profile

Relationship Graph

Source Code

benchmarks/dynamo/genai_layers/kernels.py lines 47–51

    def eager(self, args, kwargs=None) -> Any:
        if kwargs is not None:
            raise AssertionError(f"Expected kwargs to be None, but got {kwargs}")
        x, target = args
        return lambda: F.cross_entropy(x, target, reduction="none")

Domain

Subdomains

Source

Analyze Your Own Codebase

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

Try Supermodel Free