| /aosp_15_r20/external/pytorch/test/distributed/fsdp/ |
| H A D | test_fsdp_core.py | 111 sharding_strategy: Optional[ShardingStrategy],
127 sharding_strategy: Optional[ShardingStrategy],
150 sharding_strategy: Optional[ShardingStrategy],
166 sharding_strategy: Optional[ShardingStrategy],
182 sharding_strategy: Optional[ShardingStrategy],
204 sharding_strategy: Optional[ShardingStrategy],
230 sharding_strategy: Optional[ShardingStrategy],
247 sharding_strategy: Optional[ShardingStrategy],
457 sharding_strategy: ShardingStrategy,
|
| H A D | test_fsdp_use_orig_params.py | 122 sharding_strategy: ShardingStrategy,
237 self, sharding_strategy: ShardingStrategy, skip_fsdp_guards: bool
343 sharding_strategy: ShardingStrategy,
394 sharding_strategy: ShardingStrategy,
432 def _test_multiple_optimizers(self, sharding_strategy: ShardingStrategy):
564 sharding_strategy: ShardingStrategy,
634 sharding_strategy: ShardingStrategy,
689 sharding_strategy: ShardingStrategy,
755 sharding_strategy: ShardingStrategy,
1212 def _test_no_sync_correctness(self, sharding_strategy: ShardingStrategy):
[all …]
|
| H A D | test_fsdp_comm_hooks.py | 122 self, sharding_strategy: Optional[ShardingStrategy]
194 self, has_wrapping: bool, sharding_strategy: Optional[ShardingStrategy]
244 self, sharding_strategy: Optional[ShardingStrategy]
304 self, sharding_strategy: Optional[ShardingStrategy]
393 self, has_wrapping: bool, sharding_strategy: Optional[ShardingStrategy]
419 self, has_wrapping: bool, sharding_strategy: Optional[ShardingStrategy]
|
| H A D | test_fsdp_comm.py | 58 sharding_strategy: ShardingStrategy,
103 sharding_strategy: Optional[ShardingStrategy],
123 sharding_strategy: Optional[ShardingStrategy],
176 sharding_strategy: ShardingStrategy,
208 sharding_strategy: Optional[ShardingStrategy],
|
| H A D | test_fsdp_exec_order.py | 89 def wrap(sharding_strategy: ShardingStrategy, device: torch.device):
112 sharding_strategy: ShardingStrategy,
136 sharding_strategy: ShardingStrategy,
183 def test_train_eval(self, sharding_strategy: ShardingStrategy):
|
| H A D | test_fsdp_fine_tune.py | 105 sharding_strategy: ShardingStrategy,
226 sharding_strategy: ShardingStrategy,
276 sharding_strategy: ShardingStrategy,
336 sharding_strategy: ShardingStrategy,
|
| H A D | test_fsdp_hybrid_shard.py | 255 hsdp_sharding_strategy: ShardingStrategy,
347 self, hsdp_sharding_strategy: ShardingStrategy, use_orig_params: bool
395 hsdp_sharding_strategy: ShardingStrategy,
|
| H A D | test_fsdp_clip_grad_norm.py | 98 sharding_strategy: Union[ShardingStrategy, str],
264 sharding_strategy: ShardingStrategy,
|
| H A D | test_fsdp_unshard_params.py | 487 sharding_strategy: ShardingStrategy,
610 def _test_with_grads_none_grads(self, sharding_strategy: ShardingStrategy):
|
| H A D | test_fsdp_mixed_precision.py | 1008 sharding_strategy: ShardingStrategy,
1308 def _test_train_ema_eval_flow(self, sharding_strategy: ShardingStrategy):
|
| H A D | test_fsdp_grad_acc.py | 96 sharding_strategy: ShardingStrategy,
|
| H A D | test_fsdp_sharded_grad_scaler.py | 171 sharding_strategy: Optional[ShardingStrategy],
|
| H A D | test_fsdp_misc.py | 309 self, sharding_strategy: ShardingStrategy
|
| /aosp_15_r20/external/pytorch/torch/distributed/fsdp/ |
| H A D | _init_utils.py | 107 sharding_strategy: ShardingStrategy,
431 sharding_strategy: Optional[ShardingStrategy],
1189 def _get_default_comm_hook(sharding_strategy: ShardingStrategy):
|
| H A D | api.py | 32 class ShardingStrategy(Enum): class
|
| H A D | fully_sharded_data_parallel.py | 415 sharding_strategy: Optional[ShardingStrategy] = None,
|
| /aosp_15_r20/external/pytorch/test/distributed/_composable/fully_shard/ |
| H A D | test_fully_shard_runtime.py | 56 sharding_strategy: ShardingStrategy,
119 self, fsdp_wrap_mode: FSDPWrapMode, sharding_strategy: ShardingStrategy
|
| H A D | test_fully_shard_compile.py | 61 sharding_strategy: ShardingStrategy,
|
| /aosp_15_r20/external/pytorch/torch/distributed/_composable/ |
| H A D | fully_shard.py | 53 strategy: Optional[ShardingStrategy] = None,
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/tpu/ |
| H A D | tpu_embedding_configuration.proto | 60 enum ShardingStrategy { enum
|
| /aosp_15_r20/external/googleapis/google/cloud/bigquery/storage/v1beta1/ |
| H A D | storage.proto | 219 enum ShardingStrategy { enum
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/ |
| H A D | common_fsdp.py | 1294 sharding_strategy: Optional[ShardingStrategy] = None,
|
| /aosp_15_r20/prebuilts/bazel/common/remote_java_tools/java_tools/ |
| D | Runner_deploy.jar | META-INF/
META-INF/MANIFEST.MF
build-data.properties
com/
com ... |