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

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

#include <struct_info.h>

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

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

Public Member Functions
	TVM_DECLARE_BASE_OBJECT_INFO (PlacementSpecNode, 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. More...
PlacementSpecKind	kind
	The kind of placement spec. Possible values: kSharding and kReplica. More...

Static Public Attributes
static constexpr const char *	_type_key = "relax.distributed.PlacementSpec"
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_DECLARE_BASE_OBJECT_INFO()

tvm::relax::distributed::PlacementSpecNode::TVM_DECLARE_BASE_OBJECT_INFO	(	PlacementSpecNode	,
		Object
	)

Member Data Documentation

_type_key

constexpr const char* tvm::relax::distributed::PlacementSpecNode::_type_key = "relax.distributed.PlacementSpec"

staticconstexpr

_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/struct_info.h