Searched refs:get_output_amp_train_process_func (Results 1 – 4 of 4) sorted by relevance
| /external/pytorch/benchmarks/dynamo/ | ||
| D | timm_models.py | 225 def get_output_amp_train_process_func(self): member in TimmRunner |
| D | torchbench.py | 161 def get_output_amp_train_process_func(self): member in TorchBenchmarkRunner |
| D | huggingface.py | 482 def get_output_amp_train_process_func(self): member in HuggingfaceRunner |
| D | common.py | 2626 if process_fn := self.get_output_amp_train_process_func.get( |