tvm
Loading...
Searching...
No Matches
Namespaces | Classes | Enumerations
tvm::relax::distributed Namespace Reference

Namespaces

namespace  transform
 

Classes

class  DeviceMesh
 Managed reference to a DeviceMesh. More...
 
class  DeviceMeshNode
 
class  DTensorType
 Managed reference to DTensorTypeNode. More...
 
class  DTensorTypeNode
 Type of DTensor (Distributed Tensor). More...
 
class  Placement
 Managed reference to a Placement. More...
 
class  PlacementNode
 Describes how data is distributed in each dimension of the device mesh. More...
 
class  PlacementSpec
 Managed reference to PlacementSpecNode. More...
 
class  PlacementSpecNode
 Describes how data is distributed in one dimension of the device mesh. More...
 
class  ShardingNode
 

Enumerations

enum class  PlacementSpecKind : int { kSharding = 0 , kReplica = 1 }
 

Enumeration Type Documentation

◆ PlacementSpecKind

Enumerator
kSharding 
kReplica