Searched defs:forward_and_backward_pass (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/pytorch/benchmarks/dynamo/ |
| H A D | timm_models.py | 400 def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): member in TimmRunner
|
| H A D | huggingface.py | 520 def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): member in HuggingfaceRunner
|
| H A D | torchbench.py | 443 def forward_and_backward_pass(self, mod, inputs, collect_outputs=True): member in TorchBenchmarkRunner
|
| /aosp_15_r20/external/pytorch/test/inductor/ |
| H A D | test_padding.py | 83 def forward_and_backward_pass(m, inputs): function
|