Home
last modified time | relevance | path

Searched refs:_test_multiple_optimizers (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/test/distributed/fsdp/
Dtest_fsdp_use_orig_params.py429 self._test_multiple_optimizers,
432 def _test_multiple_optimizers(self, sharding_strategy: ShardingStrategy): member in TestFSDPUseOrigParamsMultipleParamGroups