Home
last modified time | relevance | path

Searched refs:tensorboard_trace_handler (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/pytorch/torch/profiler/
H A D__init__.py25 tensorboard_trace_handler,
H A Dprofiler.py457 def tensorboard_trace_handler( function
/aosp_15_r20/external/pytorch/docs/source/
H A Dprofiler.rst28 .. autofunction:: torch.profiler.tensorboard_trace_handler
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_profiler.py1090 on_trace_ready=torch.profiler.tensorboard_trace_handler(dname),
1115 on_trace_ready=torch.profiler.tensorboard_trace_handler(