Home
last modified time | relevance | path

Searched refs:function_events (Results 1 – 19 of 19) sorted by relevance

/aosp_15_r20/external/pytorch/torch/autograd/
H A Dprofiler_legacy.py51 self.function_events = None
104 self.function_events = EventList(
110 self.function_events._build_tree()
114 if self.function_events is None:
116 return repr(self.function_events)
119 if self.function_events is None:
121 return str(self.function_events)
124 if self.function_events is None:
138 assert self.function_events is not None
139 return self.function_events.table(
[all …]
H A Dprofiler.py424 def function_events(self): member in profile
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/jit/
H A Drpc_test.py1156 events = prof.function_events
1175 function_events = prof.function_events
1179 for event in function_events
1191 remote_events = [event for event in function_events if event.is_remote]
1218 function_events = prof.function_events
1220 event for event in function_events if event.name == block_scope
1232 event for event in function_events if event.name == expected_key
1258 function_events = prof.function_events
1272 evt for evt in function_events if evt.name == expected_key
1288 function_events = prof.function_events
[all …]
/aosp_15_r20/external/pytorch/torch/utils/bottleneck/
H A D__main__.py136 sorted_events = sorted(prof.function_events,
175 return sum(event.cpu_time_total for event in autograd_prof.function_events)
221 if len(autograd_prof_cpu.function_events) > 0:
/aosp_15_r20/external/pytorch/test/profiler/
H A Dtest_record_function.py56 for e in prof_result.function_events:
86 for e in prof.function_events:
143 for e in prof.function_events:
H A Dtest_profiler.py278 for e in p.function_events:
523 for e in p.function_events:
874 for e in prof.function_events:
/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Dserver_process_global_profiler.py174 self.function_events = torch.autograd.profiler_util.EventList(
179 self.function_events._build_tree()
/aosp_15_r20/external/pytorch/test/jit/
H A Dtest_with.py635 function_events = p.function_events
637 rf_events = [evt for evt in function_events if evt.name == "foo"]
644 evt for evt in function_events if evt.name == "nested"
H A Dtest_torchbind.py420 for e in prof.function_events:
/aosp_15_r20/external/pytorch/torch/testing/_internal/
H A Ddist_utils.py189 def get_function_event(function_events, partial_event_name): argument
199 …event = [event for event in function_events if partial_event_name in event.name][0] # noqa: RUF015
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py1781 function_events = prof.function_events
1782 remote_events = [event for event in function_events if event.is_remote]
1791 local_function_events = prof.function_events
1808 function_events = p.function_events
1809 event_cpu_mem_usages = {event.cpu_memory_usage for event in function_events}
1818 function_events = p.function_events
1819 event_cpu_mem_usages = {event.cpu_memory_usage for event in function_events}
1832 events = p.function_events
1859 events = prof.function_events
1925 events = prof.function_events
[all …]
H A Ddist_autograd_test.py1460 function_events = p.function_events
1463 return next(event for event in function_events if partial_key in event.name)
/aosp_15_r20/external/pytorch/test/dynamo/
H A Dtest_profiler.py118 (torch.autograd.profiler.profile, lambda prof: prof.function_events),
/aosp_15_r20/external/pytorch/torch/profiler/
H A Dprofiler.py294 return self.profiler.function_events
/aosp_15_r20/external/pytorch/test/
H A Dtest_autograd.py4658 for evt in p.function_events:
4680 for e in p.function_events:
4744 for info in p.function_events:
4764 function_events = p.function_events
4765 foo_event = next(event for event in function_events if "foo" in event.name)
4779 function_events = p.function_events
4780 foo_event = next(event for event in function_events if "bar" in event.name)
4868 print(prof.function_events)
4876 for event in prof.function_events:
4944 events = p.function_events
H A Dtest_jit.py393 for e in prof.function_events:
1835 return [e.name for e in prof.function_events]
1872 return [e.name for e in prof.function_events]
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_cpu_repro.py2639 for e in prof.profiler.function_events:
H A Dtest_torchinductor.py8908 "inductor_wrapper_call" in e.name for e in prof.profiler.function_events
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/
H A Ddistributed_test.py202 else profiler.function_events