Zoom in maybe?

image.png1501×348 26.2 KB

This is what I see (code used to generate it below)
import torch

x = torch.randn((1, 1), requires_grad=True)
y = torch.randn((1, 1), requires_grad=True)

with torch.autograd.profiler.profile() as prof:
    z = x.matmul(y)
    z.sum().backward()

prof.export_chrome_trace("prof.out")