Searched refs:_run_and_format_data_flow (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/test/profiler/ |
| H A D | test_memory_profiler.py | 319 def _run_and_format_data_flow( member in TestDataFlow
451 self._run_and_format_data_flow(inputs, f),
473 self._run_and_format_data_flow({"x": torch.ones((1,))}, f),
494 self._run_and_format_data_flow(inputs, f0),
506 self._run_and_format_data_flow(inputs, f0),
529 self._run_and_format_data_flow(inputs, f0),
538 self._run_and_format_data_flow(inputs, f1),
552 self._run_and_format_data_flow(
583 self._run_and_format_data_flow({}, f),
630 self._run_and_format_data_flow(inputs, f_fwd),
[all …]
|