struct_info.h File Reference

Struct info for DTensor (Distributed Tensor) More...

#include <tvm/relax/distributed/global_info.h>
#include <tvm/relax/struct_info.h>

Include dependency graph for struct_info.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
class	tvm::relax::distributed::PlacementSpecNode
	Describes how data is distributed in one dimension of the device mesh. More...
class	tvm::relax::distributed::PlacementSpec
	Managed reference to PlacementSpecNode. More...
class	tvm::relax::distributed::ShardingNode
class	tvm::relax::distributed::PlacementNode
	Describes how data is distributed in each dimension of the device mesh. More...
class	tvm::relax::distributed::Placement
	Managed reference to a Placement. More...
class	tvm::relax::distributed::DTensorStructInfoNode
	StructInfo of DTensor (Distributed Tensor). More...
class	tvm::relax::distributed::DTensorStructInfo
	Managed reference to DTensorStructInfoNode. More...

Namespaces
	tvm
	runtime implementation for LibTorch/TorchScript.
	tvm::relax
	tvm::relax::distributed

Enumerations
enum class	tvm::relax::distributed::PlacementSpecKind : int { tvm::relax::distributed::kSharding = 0 , tvm::relax::distributed::kReplica = 1 }

Detailed Description

Struct info for DTensor (Distributed Tensor)