distributed.checkpoint (path) in projects: external

Searched +path:distributed +path:. +path:checkpoint (Results 1 – 25 of 59) sorted by relevance

/external/pytorch/test/distributed/checkpoint/
D	test_tp_checkpoint.py
D	test_nested_dict.py
D	test_fsdp_tp_checkpoint_conversion.py
D	test_fsdp_optim_state.py
D	test_state_dict.py
D	test_dtensor_resharding.py
D	test_fsdp_model_state.py
D	test_file_system_checkpoint_cpu.py
D	test_dedup_tensors.py
D	test_format_utils.py
D	test_fsspec.py
/external/pytorch/torch/distributed/checkpoint/
D	api.py
D	filesystem.py
D	state_dict_loader.py
D	_fsspec_filesystem.py
D	optimizer.py
D	planner.py
D	_sharded_tensor_utils.py
D	staging.py
D	state_dict.py
D	_nested_dict.py
D	format_utils.py
D	_version.py
D	storage.py
D	_checkpointer.py