tvm
|
Managed reference to a DeviceMesh. More...
#include <global_info.h>
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) | |
![]() | |
TVM_DEFINE_OBJECT_REF_METHODS (GlobalInfo, ObjectRef, GlobalInfoNode) | |
Managed reference to a DeviceMesh.
tvm::relax::distributed::DeviceMesh::DeviceMesh | ( | ffi::Shape | shape, |
Array< Integer > | device_ids | ||
) |
tvm::relax::distributed::DeviceMesh::DeviceMesh | ( | ffi::Shape | shape, |
Range | device_range | ||
) |
tvm::relax::distributed::DeviceMesh::TVM_DEFINE_OBJECT_REF_METHODS | ( | DeviceMesh | , |
GlobalInfo | , | ||
DeviceMeshNode | |||
) |