Searched defs:_ProfilerEvent (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/profiler/ |
| H A D | _pattern_matcher.py | 45 def report(self, event: _ProfilerEvent):
58 def summary(self, events: List[_ProfilerEvent]):
69 def benchmark_summary(self, events: List[_ProfilerEvent]):
90 def match(self, event: _ProfilerEvent):
106 def root_of(self, event: _ProfilerEvent):
111 def siblings_of(self, event: _ProfilerEvent):
119 def next_of(self, event: _ProfilerEvent):
123 def prev_of(self, event: _ProfilerEvent):
127 def go_up_until(self, event: _ProfilerEvent, predicate):
144 def match(self, event: _ProfilerEvent):
[all …]
|
| H A D | _memory_profiler.py | 153 node: _ProfilerEvent,
194 def extract_parameters(node: _ProfilerEvent) -> Iterator[TensorKey]:
201 node: _ProfilerEvent,
208 def get_scopes(event: Optional[_ProfilerEvent]) -> Tuple[RecordScope, ...]:
421 def __init__(self, event: _ProfilerEvent, graph: "DataFlowGraph") -> None:
587 def leaf_op(e: _ProfilerEvent) -> bool:
593 def children_fn(e: _ProfilerEvent):
|