Searched defs:pick_grad (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/benchmarks/dynamo/ | ||
| D | timm_models.py | 349 def pick_grad(self, name, is_training): member in TimmRunner |
| D | huggingface.py | 485 def pick_grad(self, name, is_training): member in HuggingfaceRunner |
| D | torchbench.py | 399 def pick_grad(self, name, is_training): member in TorchBenchmarkRunner |