torch.autograd.profiler.load_nvprof

torch.autograd.profiler.load_nvprof(path) [source]

Open an nvprof trace file and parses autograd annotations.

Parameters

path (str) – path to nvprof trace