tvm
Public Member Functions | Static Public Member Functions | Public Attributes | Static Public Attributes | List of all members
tvm::relax::distributed::DTensorStructInfoNode Class Reference

StructInfo of DTensor (Distributed Tensor). More...

#include <struct_info.h>

Inheritance diagram for tvm::relax::distributed::DTensorStructInfoNode:
Collaboration diagram for tvm::relax::distributed::DTensorStructInfoNode:

Public Member Functions

 TVM_DECLARE_FINAL_OBJECT_INFO (DTensorStructInfoNode, StructInfoNode)
 
- Public Member Functions inherited from tvm::relax::StructInfoNode
 TVM_DECLARE_BASE_OBJECT_INFO (StructInfoNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 
- Static Public Member Functions inherited from tvm::relax::StructInfoNode
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...
 
- Public Attributes inherited from tvm::relax::StructInfoNode
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 Public Attributes inherited from tvm::relax::StructInfoNode
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
 

Detailed Description

StructInfo of DTensor (Distributed Tensor).

Member Function Documentation

◆ RegisterReflection()

static void tvm::relax::distributed::DTensorStructInfoNode::RegisterReflection ( )
inlinestatic

◆ TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::distributed::DTensorStructInfoNode::TVM_DECLARE_FINAL_OBJECT_INFO ( DTensorStructInfoNode  ,
StructInfoNode   
)

Member Data Documentation

◆ _type_key

constexpr const char* tvm::relax::distributed::DTensorStructInfoNode::_type_key = "relax.DTensorStructInfo"
staticconstexpr

◆ device_mesh

DeviceMesh tvm::relax::distributed::DTensorStructInfoNode::device_mesh

The device mesh of the tensor.

◆ placement

Placement tvm::relax::distributed::DTensorStructInfoNode::placement

The placement of the tensor among the device mesh.

◆ tensor_sinfo

TensorStructInfo tvm::relax::distributed::DTensorStructInfoNode::tensor_sinfo

The struct info inherited from TensorStructInfo.


The documentation for this class was generated from the following file: