tvm::relax::distributed::DeviceMeshNode Class Reference

#include <global_info.h>

Inheritance diagram for tvm::relax::distributed::DeviceMeshNode:

Collaboration diagram for tvm::relax::distributed::DeviceMeshNode:

Public Member Functions
	TVM_DECLARE_FINAL_OBJECT_INFO (DeviceMeshNode, GlobalInfoNode)
Public Member Functions inherited from tvm::GlobalInfoNode
	TVM_DECLARE_BASE_OBJECT_INFO (GlobalInfoNode, Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
ffi::Shape	shape
	logical shape of the mesh More...
Array< Integer >	device_ids
	device ids in the mesh More...
Optional< Range >	device_range
	Optionally use range to represent device_ids. More...

Static Public Attributes
static constexpr const char *	_type_key = "relax.distributed.DeviceMesh"
Static Public Attributes inherited from tvm::GlobalInfoNode
static constexpr const char *	_type_key = "ir.GlobalInfo"
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindTreeNode

Member Function Documentation

RegisterReflection()

static void tvm::relax::distributed::DeviceMeshNode::RegisterReflection ( )

inlinestatic

TVM_DECLARE_FINAL_OBJECT_INFO()

tvm::relax::distributed::DeviceMeshNode::TVM_DECLARE_FINAL_OBJECT_INFO	(	DeviceMeshNode	,
		GlobalInfoNode
	)

Member Data Documentation

_type_key

constexpr const char* tvm::relax::distributed::DeviceMeshNode::_type_key = "relax.distributed.DeviceMesh"

staticconstexpr

device_ids

Array<Integer> tvm::relax::distributed::DeviceMeshNode::device_ids

device ids in the mesh

device_range

Optional<Range> tvm::relax::distributed::DeviceMeshNode::device_range

Optionally use range to represent device_ids.

shape

ffi::Shape tvm::relax::distributed::DeviceMeshNode::shape

logical shape of the mesh

---

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

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