tvm::relax::distributed::PlacementNode Class Reference

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

#include <type.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, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Array< PlacementSpec >	dim_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	,
		ffi::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:

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