tvm::relax::distributed::PlacementSpecNode Class Reference

Describes how data is distributed in one dimension of the device mesh. More...

#include <type.h>

Inheritance diagram for tvm::relax::distributed::PlacementSpecNode:

Collaboration diagram for tvm::relax::distributed::PlacementSpecNode:

Public Member Functions
	TVM_FFI_DECLARE_OBJECT_INFO ("relax.distributed.PlacementSpec", PlacementSpecNode, ffi::Object)

Static Public Member Functions
static void	RegisterReflection ()

Public Attributes
int	axis
	If the kind is sharding, this value represents the tensor dimension to shard. otherwise, axis is -1.
PlacementSpecKind	kind
	The kind of placement spec. Possible values: kSharding and kReplica.

Static Public Attributes
static constexpr TVMFFISEqHashKind	_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode

Detailed Description

Describes how data is distributed in one dimension of the device mesh.

Member Function Documentation

RegisterReflection()

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

inlinestatic

TVM_FFI_DECLARE_OBJECT_INFO()

tvm::relax::distributed::PlacementSpecNode::TVM_FFI_DECLARE_OBJECT_INFO	(	"relax.distributed.PlacementSpec"	,
		PlacementSpecNode	,
		ffi::Object
	)

Member Data Documentation

_type_s_eq_hash_kind

constexpr TVMFFISEqHashKind tvm::relax::distributed::PlacementSpecNode::_type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode

staticconstexpr

axis

int tvm::relax::distributed::PlacementSpecNode::axis

If the kind is sharding, this value represents the tensor dimension to shard. otherwise, axis is -1.

kind

PlacementSpecKind tvm::relax::distributed::PlacementSpecNode::kind

The kind of placement spec. Possible values: kSharding and kReplica.

---

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

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