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