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
String	ToString () const
	TVM_DECLARE_FINAL_OBJECT_INFO (PlacementNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
Array< PlacementSpec >	dim_specs
	specs for each dim of device mesh. More...

Static Public Attributes
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode
static constexpr const char *	_type_key = "relax.distributed.Placement"

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()

String tvm::relax::distributed::PlacementNode::ToString

(

)

const

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::distributed::PlacementNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	PlacementNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::relax::distributed::PlacementNode::_type_key = "relax.distributed.Placement"

staticconstexpr

_type_s_eq_hash_kind

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

staticconstexpr

dim_specs

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:

• include/tvm/relax/distributed/struct_info.h