torch.profiler

若API“是否支持”“是”“限制与说明”为空,说明此API和原生API支持度保持一致。

API名称

是否支持

限制与说明

torch.profiler._KinetoProfile

  

torch.profiler._KinetoProfile.add_metadata

  

torch.profiler._KinetoProfile.add_metadata_json

  

torch.profiler._KinetoProfile.events

  

torch.profiler._KinetoProfile.export_chrome_trace

  

torch.profiler._KinetoProfile.export_memory_timeline

  

torch.profiler._KinetoProfile.export_stacks

  

torch.profiler._KinetoProfile.key_averages

  

torch.profiler.profile

  

torch.profiler.profile.step

  

torch.profiler.ProfilerAction

  

torch.profiler.ProfilerActivity

  

torch.profiler.ProfilerActivity.name

  

torch.profiler.schedule

  

torch.profiler.tensorboard_trace_handler

  

torch.profiler.itt.is_available

  

torch.profiler.itt.mark

  

torch.profiler.itt.range_push

  

torch.profiler.itt.range_pop