distributed.tensor.parallel (path) in projects: external

Searched +path:distributed +path:. +path:tensor +path:. +path:parallel (Results 1 – 16 of 16) sorted by relevance

/external/pytorch/torch/distributed/tensor/parallel/
D	input_reshard.py
D	fsdp.py
D	__init__.py
D	api.py
D	loss.py
D	_data_parallel_utils.py
D	_utils.py
D	ddp.py
D	style.py
/external/pytorch/test/distributed/tensor/parallel/
D	test_tp_random_state.py
D	test_tp_style.py
D	test_micro_pipeline_tp.py
D	test_parallelize_api.py
D	test_tp_examples.py
D	__init__.py
/external/pytorch/docs/source/
D	distributed.tensor.parallel.rst