tvm
|
StructInfo of DTensor (Distributed Tensor). More...
#include <struct_info.h>
Public Member Functions | |
TVM_DECLARE_FINAL_OBJECT_INFO (DTensorStructInfoNode, StructInfoNode) | |
![]() | |
TVM_DECLARE_BASE_OBJECT_INFO (StructInfoNode, Object) | |
Static Public Member Functions | |
static void | RegisterReflection () |
![]() | |
static void | RegisterReflection () |
Public Attributes | |
TensorStructInfo | tensor_sinfo |
The struct info inherited from TensorStructInfo. More... | |
DeviceMesh | device_mesh |
The device mesh of the tensor. More... | |
Placement | placement |
The placement of the tensor among the device mesh. More... | |
![]() | |
Span | span |
Span that points to the original source code. Reserved debug information. More... | |
Static Public Attributes | |
static constexpr const char * | _type_key = "relax.DTensorStructInfo" |
![]() | |
static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
static constexpr const char * | _type_key = "ir.StructInfo" |
static constexpr const uint32_t | _type_child_slots = 7 |
StructInfo of DTensor (Distributed Tensor).
|
inlinestatic |
tvm::relax::distributed::DTensorStructInfoNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | DTensorStructInfoNode | , |
StructInfoNode | |||
) |
|
staticconstexpr |
DeviceMesh tvm::relax::distributed::DTensorStructInfoNode::device_mesh |
The device mesh of the tensor.
Placement tvm::relax::distributed::DTensorStructInfoNode::placement |
The placement of the tensor among the device mesh.
TensorStructInfo tvm::relax::distributed::DTensorStructInfoNode::tensor_sinfo |
The struct info inherited from TensorStructInfo.