Searched refs:_zero_model (Results 1 – 3 of 3) sorted by relevance
| /external/pytorch/test/distributed/_composable/fully_shard/ |
| D | test_fully_shard_model_checkpoint.py | 23 _zero_model,
70 _zero_model(load_composable, summon_full=False)
102 _zero_model(load_composable, summon_full=False)
160 _zero_model(load_model, zero_buffers=True, summon_full=False)
199 _zero_model(load_composable, summon_full=False)
|
| /external/pytorch/test/distributed/fsdp/ |
| D | test_fsdp_state_dict.py | 48 _zero_model,
361 _zero_model(model_new)
451 _zero_model(new_model, zero_buffers=True)
490 _zero_model(new_model, zero_buffers=True)
610 _zero_model(model_new, zero_buffers=True)
677 _zero_model(model_new, zero_buffers=True)
741 _zero_model(model)
810 _zero_model(blank_model)
981 _zero_model(new_fsdp)
|
| /external/pytorch/torch/testing/_internal/ |
| D | common_fsdp.py | 142 def _zero_model( function
1277 _zero_model(model)
|