Searched defs:run_backward (Results 1 – 7 of 7) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 100 def run_backward(self, loss) -> None: member in FSDPTestModel
299 def run_backward(self, loss): member in TransformerWithSharedParams
428 def run_backward(self, loss): member in NestedWrappedModule
635 def run_backward(self, loss): member in ModuleWithDelay
783 def run_backward(self, loss): member in MixtureOfExperts
|
| /aosp_15_r20/external/pytorch/benchmarks/operator_benchmark/ |
| H A D | benchmark_core.py | 50 bench_op_obj, orig_test_attrs, tags, OperatorTestCase, run_backward, bwd_input argument
75 configs, bench_op, OperatorTestCase, run_backward, op_name_function=None argument
|
| H A D | benchmark_pytorch.py | 174 def run_backward(self, num_runs, print_per_iter=False): member in PyTorchOperatorTestCase
|
| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_exec_order.py | 72 def run_backward(self, loss): member in Model
|
| H A D | test_fsdp_ignored_modules.py | 69 def run_backward(self, loss): member in Model
|
| H A D | test_fsdp_optim_state.py | 171 def run_backward(self, loss): member in NestedModel
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd.cpp | 90 static variable_list run_backward( in run_backward() function
|