TPConfig#

class composer.utils.TPConfig(device_mesh=None, tensor_parallel_degree=1, layer_plan=None)[source]#

Configuration for tensor parallelism (TP).