torch.distributed.checkpoint (path) in projects: external

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

/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
D	_dedup_tensors.py
D	metadata.py
D	utils.py
D	_traverse.py
D	_dedup_save_plans.py
D	stateful.py
D	logger.py
D	state_dict_saver.py
D	logging_handlers.py
D	__init__.py
D	default_planner.py