MemRecordsAcc

torch.autograd.profiler_util.MemRecordsAcc(mem_records)[源代码]

用于在区间内访问 mem_records 的加速结构。

in_interval(start_us, end_us)[源代码]

返回给定区间内的所有记录,并为了保持向后兼容性,将函数中的微秒(us)转换为纳秒(ns)

本页目录