|
tvm
|
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 } |