tvm::relax::distributed::DeviceMesh Class Reference

Managed reference to a DeviceMesh. More...

#include <global_info.h>

Inheritance diagram for tvm::relax::distributed::DeviceMesh:

Collaboration diagram for tvm::relax::distributed::DeviceMesh:

Public Member Functions
	DeviceMesh (ffi::Shape shape, ffi::Array< Integer > device_ids)
	DeviceMesh (ffi::Shape shape, Range device_range)
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (DeviceMesh, GlobalInfo, DeviceMeshNode)
Public Member Functions inherited from tvm::GlobalInfo
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE (GlobalInfo, ObjectRef, GlobalInfoNode)

Detailed Description

Managed reference to a DeviceMesh.

See also

DeviceMeshNode

Constructor & Destructor Documentation

DeviceMesh() [1/2]

tvm::relax::distributed::DeviceMesh::DeviceMesh	(	ffi::Shape	shape,
		ffi::Array< Integer >	device_ids
	)

DeviceMesh() [2/2]

tvm::relax::distributed::DeviceMesh::DeviceMesh	(	ffi::Shape	shape,
		Range	device_range
	)

Member Function Documentation

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE()

tvm::relax::distributed::DeviceMesh::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NULLABLE	(	DeviceMesh	,
		GlobalInfo	,
		DeviceMeshNode
	)

---

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

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