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