torch.profiler

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

API名称

是否支持

限制与说明

torch.profiler._KinetoProfile

  

torch.profiler._KinetoProfile.add_metadata

  

torch.profiler._KinetoProfile.add_metadata_json

  

torch.profiler._KinetoProfile.export_chrome_trace

  

torch.profiler._KinetoProfile.export_stacks

  

torch.profiler.profile

  

torch.profiler.profile.step

  

torch.profiler.ProfilerAction

  

torch.profiler.ProfilerActivity

  

torch.profiler.schedule

  

torch.profiler.tensorboard_trace_handler