Searched refs:FunctionEvent (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/torch/autograd/ |
| D | profiler_util.py | 117 current_events: List[FunctionEvent] = []
443 class FunctionEvent(FormattedTimesMixin): class
483 self.cpu_children: List[FunctionEvent] = []
484 self.cpu_parent: Optional[FunctionEvent] = None
519 assert isinstance(child, FunctionEvent)
531 assert isinstance(parent, FunctionEvent)
668 self.cpu_children: Optional[List[FunctionEvent]] = None
669 self.cpu_parent: Optional[FunctionEvent] = None
693 assert isinstance(other, (FunctionEvent, FunctionEventAvg))
|
| D | profiler_legacy.py | 20 FunctionEvent,
249 fe = FunctionEvent(
292 fe = FunctionEvent(
|
| D | profiler.py | 31 FunctionEvent,
541 device_corr_map: Dict[int, List[FunctionEvent]] = {}
565 fe = FunctionEvent(
644 fe = FunctionEvent(
1041 evt = FunctionEvent(
|
| /external/pytorch/test/ |
| D | test_autograd.py | 46 FunctionEvent,
4802 FunctionEvent(
4842 FunctionEvent(id=0, node_id=0, name="foo", thread=0, start_us=10, end_us=15)
4845 FunctionEvent(id=1, node_id=0, name="foo", thread=0, start_us=20, end_us=30)
|