tvm
|
StructInfo of DTensor (Distributed Tensor). More...
#include <struct_info.h>
Public Member Functions | |
TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.DTensorStructInfo", DTensorStructInfoNode, StructInfoNode) | |
![]() | |
TVM_FFI_DECLARE_OBJECT_INFO ("ir.StructInfo", 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... | |
Additional Inherited Members | |
![]() | |
static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode |
static constexpr const uint32_t | _type_child_slots = 7 |
StructInfo of DTensor (Distributed Tensor).
|
inlinestatic |
tvm::relax::distributed::DTensorStructInfoNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "relax.DTensorStructInfo" | , |
DTensorStructInfoNode | , | ||
StructInfoNode | |||
) |
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.