type.h File Reference

Type definitions for DTensor (Distributed Tensor) More...

#include <tvm/relax/distributed/global_info.h>
#include <tvm/relax/type.h>
#include <utility>

Include dependency graph for type.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::DTensorTypeNode
	Type of DTensor (Distributed Tensor). More...
class	tvm::relax::distributed::DTensorType
	Managed reference to DTensorTypeNode. More...

Namespaces
	tvm
	An object that builds and maintains block scope and StmtSref mapping for Dependence analysis.
	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

Type definitions for DTensor (Distributed Tensor)