Home
last modified time | relevance | path

Searched defs:DDP (Results 1 – 25 of 26) sorted by relevance

12

/external/pytorch/benchmarks/distributed/rpc/parameter_server/trainer/
Dddp_models.py1 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/test/distributed/fsdp/
Dtest_fsdp_use_orig_params.py30 from torch.nn.parallel.distributed import DistributedDataParallel as DDP unknown
166 ddp_model: DDP,
199 def _check_ddp_fsdp_param_parity(self, ddp_model: DDP, fsdp_model: FSDP):
785 ddp_model: DDP, fsdp_model: FSDP, between_fwd_and_bwd: bool
884 def _check_param_parity(self, ddp_model: DDP, fsdp_model: FSDP):
Dtest_fsdp_unshard_params.py22 from torch.nn.parallel.distributed import DistributedDataParallel as DDP unknown
491 ddp_model: DDP,
Dtest_fsdp_sharded_grad_scaler.py21 from torch.nn.parallel.distributed import DistributedDataParallel as DDP unknown
Dtest_fsdp_clip_grad_norm.py18 from torch.nn.parallel import DistributedDataParallel as DDP unknown
Dtest_fsdp_fine_tune.py16 from torch.nn.parallel import DistributedDataParallel as DDP unknown
Dtest_fsdp_comm.py16 from torch.nn.parallel.distributed import DistributedDataParallel as DDP unknown
Dtest_fsdp_misc.py32 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/benchmarks/dynamo/
Ddistributed.py10 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/test/distributed/_composable/test_composability/
Dtest_pp_composability.py23 from torch.nn.parallel import DistributedDataParallel as DDP unknown
Dtest_2d_composability.py40 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/torch/distributed/benchmarks/
Dbenchmark_ddp_rpc.py24 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/torch/distributed/_composable/
Dreplicate.py152 class DDP: class
/external/pytorch/test/distributed/
Dtest_c10d_pypg.py13 from torch.nn.parallel import DistributedDataParallel as DDP unknown
Dtest_dynamo_distributed.py35 from torch.nn.parallel import DistributedDataParallel as DDP unknown
1095 from torch.nn.parallel import DistributedDataParallel as DDP unknown
1106 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/test/distributed/_composable/
Dtest_replicate_with_compiler.py28 from torch.nn.parallel.distributed import DistributedDataParallel as DDP unknown
/external/pytorch/test/distributed/checkpoint/e2e/
Dtest_e2e_save_and_load.py89 DDP = auto() variable in ModelType
/external/pytorch/test/dynamo/
Dtest_structured_trace.py21 from torch.nn.parallel import DistributedDataParallel as DDP unknown
Dtest_logging.py14 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/test/inductor/
Dtest_layout_optim.py97 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/test/distributed/checkpoint/
Dtest_state_dict.py40 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/rust/android-crates-io/crates/etherparse/src/net/
Dip_number_impl.rs125 pub const DDP: IpNumber = Self(37); constant
848 pub const DDP: IpNumber = IpNumber::DDP; // 37 constant
/external/pytorch/torch/distributed/checkpoint/
Dstate_dict.py54 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/test/distributed/optim/
Dtest_zero_redundancy_optimizer.py32 from torch.nn.parallel import DistributedDataParallel as DDP unknown
/external/pytorch/torch/testing/_internal/
Dcommon_fsdp.py56 from torch.nn.parallel.distributed import DistributedDataParallel as DDP unknown

12