tvm
Public Member Functions | List of all members
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, Array< Integer > device_ids)
 
 DeviceMesh (ffi::Shape shape, Range device_range)
 
 TVM_DEFINE_OBJECT_REF_METHODS (DeviceMesh, GlobalInfo, DeviceMeshNode)
 
- Public Member Functions inherited from tvm::GlobalInfo
 TVM_DEFINE_OBJECT_REF_METHODS (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,
Array< Integer device_ids 
)

◆ DeviceMesh() [2/2]

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

Member Function Documentation

◆ TVM_DEFINE_OBJECT_REF_METHODS()

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

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