#include <global_info.h>
◆ RegisterReflection()
static void tvm::relax::distributed::DeviceMeshNode::RegisterReflection |
( |
| ) |
|
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()
tvm::relax::distributed::DeviceMeshNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL |
( |
"relax.distributed.DeviceMesh" |
, |
|
|
DeviceMeshNode |
, |
|
|
GlobalInfoNode |
|
|
) |
| |
◆ device_ids
ffi::Array<Integer> tvm::relax::distributed::DeviceMeshNode::device_ids |
◆ device_range
ffi::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: