Searched defs:get_output_amp_train_process_func (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ | ||
| H A D | timm_models.py | 225 def get_output_amp_train_process_func(self): member in TimmRunner |
| H A D | huggingface.py | 482 def get_output_amp_train_process_func(self): member in HuggingfaceRunner |
| H A D | torchbench.py | 161 def get_output_amp_train_process_func(self): member in TorchBenchmarkRunner |