Searched defs:static_graph (Results 1 – 9 of 9) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/nn/parallel/ |
| H A D | distributed.py | 640 static_graph=False, argument
1137 static_graph, argument
|
| /aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
| H A D | logger.cpp | 179 bool static_graph) { in set_construction_data_and_log()
|
| /aosp_15_r20/external/pytorch/test/distributed/ |
| H A D | test_dynamo_distributed.py | 505 def _test_hf_bert_ddp_inductor(self, static_graph): argument
527 def _test_hf_bert_aot_eager(self, static_graph): argument
|
| H A D | test_c10d_ucc.py | 411 self, gradient_as_bucket_view=False, static_graph=False argument
|
| H A D | test_c10d_nccl.py | 1749 static_graph=False, argument
1783 self, gradient_as_bucket_view=False, static_graph=False argument
|
| H A D | test_c10d_gloo.py | 1552 self, gradient_as_bucket_view=False, static_graph=False argument
|
| H A D | test_c10d_common.py | 411 static_graph=False, argument
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/ |
| H A D | distributed_test.py | 4295 static_graph=False, argument
4554 static_graph, argument
7641 def _test_ddp_ignore_params_arg(self, static_graph=False): argument
8693 self, find_unused_parameters=False, static_graph=False argument
9439 def _test_ddp_new_tensor_in_fwd(self, static_graph): argument
|
| /aosp_15_r20/external/pytorch/test/distributed/optim/ |
| H A D | test_zero_redundancy_optimizer.py | 1251 static_graph, argument
|