Rate this Page

MemRecordsAcc#

class torch.autograd.profiler_util.MemRecordsAcc(mem_records)[source]#

Acceleration structure for accessing mem_records in interval.

in_interval(start_us, end_us)[source]#

Return all records in the given interval To maintain backward compatibility, convert us to ns in function