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

Describes how data is distributed in each dimension of the device mesh. More...

#include <struct_info.h>

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

Public Member Functions

ffi::String ToString () const
 
 TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.distributed.Placement", PlacementNode, Object)
 

Static Public Member Functions

static void RegisterReflection ()
 

Public Attributes

ffi::Array< PlacementSpecdim_specs
 specs for each dim of device mesh. More...
 

Static Public Attributes

static constexpr TVMFFISEqHashKind _type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode
 

Detailed Description

Describes how data is distributed in each dimension of the device mesh.

Member Function Documentation

◆ RegisterReflection()

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

◆ ToString()

ffi::String tvm::relax::distributed::PlacementNode::ToString ( ) const

◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()

tvm::relax::distributed::PlacementNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL ( "relax.distributed.Placement"  ,
PlacementNode  ,
Object   
)

Member Data Documentation

◆ _type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::relax::distributed::PlacementNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode
staticconstexpr

◆ dim_specs

ffi::Array<PlacementSpec> tvm::relax::distributed::PlacementNode::dim_specs

specs for each dim of device mesh.


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