Managed reference to DTensorStructInfoNode. More...
#include <struct_info.h>
Inheritance diagram for tvm::relax::distributed::DTensorStructInfo:
Collaboration diagram for tvm::relax::distributed::DTensorStructInfo:
Public Member Functions | |
| DTensorStructInfo (TensorStructInfo tensor_sinfo, DeviceMesh device_mesh, Placement placement, Span span=Span()) | |
| Construction with device mesh and placement. More... | |
| TVM_DEFINE_OBJECT_REF_METHODS (DTensorStructInfo, StructInfo, DTensorStructInfoNode) | |
 Public Member Functions inherited from tvm::relax::StructInfo | |
| TVM_DEFINE_OBJECT_REF_METHODS (StructInfo, ObjectRef, StructInfoNode) | |
Detailed Description
Managed reference to DTensorStructInfoNode.
- See also
- DTensorStructInfoNode
Constructor & Destructor Documentation
◆ DTensorStructInfo()
| tvm::relax::distributed::DTensorStructInfo::DTensorStructInfo | ( | TensorStructInfo | tensor_sinfo, |
| DeviceMesh | device_mesh, | ||
| Placement | placement, | ||
| Span | span = Span() |
||
| ) |
Construction with device mesh and placement.
- Parameters
-
tensor_sinfo The struct info inherited from TensorStructInfo device_mesh The device mesh of the tensor. placement The placement of the tensor among the device mesh. span The span of the AST.
Member Function Documentation
◆ TVM_DEFINE_OBJECT_REF_METHODS()
| tvm::relax::distributed::DTensorStructInfo::TVM_DEFINE_OBJECT_REF_METHODS | ( | DTensorStructInfo | , |
| StructInfo | , | ||
| DTensorStructInfoNode | |||
| ) |
The documentation for this class was generated from the following file:
- include/tvm/relax/distributed/struct_info.h
