Home
last modified time | relevance | path

Searched refs:collect_results (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/pytorch/torch/utils/_strobelight/
H A Dcli_function_profiler.py216 collect_results: bool,
225 if not collect_results:
246 self._stop_strobelight_no_throw(collect_results=False)
274 self._stop_strobelight_no_throw(collect_results=True)
279 self._stop_strobelight_no_throw(collect_results=False)
/aosp_15_r20/external/pytorch/torch/_strobelight/
H A Dcli_function_profiler.py220 collect_results: bool,
229 if not collect_results:
250 self._stop_strobelight_no_throw(collect_results=False)
283 self._stop_strobelight_no_throw(collect_results=True)
288 self._stop_strobelight_no_throw(collect_results=False)
/aosp_15_r20/external/libopus/dnn/torch/lpcnet/scripts/
H A Dcollect_multi_run_results.py58 def collect_results(folder): function
175 results[uuid] = collect_results(os.path.join(args.input, uuid))
/aosp_15_r20/external/pytorch/benchmarks/dynamo/
H A Dtimm_models.py18 from torch._dynamo.testing import collect_results, reduce_to_scalar_loss
411 return collect_results(mod, pred, loss, cloned_inputs)
H A Dtorchbench.py21 from torch._dynamo.testing import collect_results, reduce_to_scalar_loss
455 return collect_results(mod, pred, loss, cloned_inputs)
H A Dhuggingface.py30 from torch._dynamo.testing import collect_results
529 return collect_results(mod, pred, loss, cloned_inputs)
/aosp_15_r20/external/pytorch/test/distributed/
H A Dtest_dynamo_distributed.py26 from torch._dynamo.testing import collect_results
302 correct_results = collect_results(
305 opt_results = collect_results(opt_model, opt_outputs.logits, opt_loss, inputs_flat)
714 correct_results = collect_results(
717 opt_results = collect_results(
763 correct_results = collect_results(
766 opt_results = collect_results(
/aosp_15_r20/external/pytorch/torch/_dynamo/
H A Ddebug_utils.py316 from .testing import collect_results, reduce_to_scalar_loss, requires_bwd_pass
333 return collect_results(gm, out, None, args)
H A Dtesting.py53 def collect_results(model, prediction, loss, example_inputs): function