tvm::relax::distributed::PlacementSpec Class Reference

Managed reference to PlacementSpecNode. More...

#include <type.h>

Inheritance diagram for tvm::relax::distributed::PlacementSpec:

Collaboration diagram for tvm::relax::distributed::PlacementSpec:

Public Member Functions
	PlacementSpec (ffi::UnsafeInit tag, ffi::ObjectPtr< PlacementSpecNode > n)
	TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE (PlacementSpec, ffi::ObjectRef, PlacementSpecNode)

Static Public Member Functions
static PlacementSpec	Sharding (int axis)
static PlacementSpec	Replica ()

Detailed Description

Managed reference to PlacementSpecNode.

See also

PlacementSpecNode

Constructor & Destructor Documentation

PlacementSpec()

tvm::relax::distributed::PlacementSpec::PlacementSpec ( ffi::UnsafeInit  tag, ffi::ObjectPtr< PlacementSpecNode >  n  )

inlineexplicit

Member Function Documentation

Replica()

static PlacementSpec tvm::relax::distributed::PlacementSpec::Replica ( )

static

Sharding()

static PlacementSpec tvm::relax::distributed::PlacementSpec::Sharding ( int  axis )

static

TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE()

tvm::relax::distributed::PlacementSpec::TVM_FFI_DEFINE_OBJECT_REF_METHODS_NOTNULLABLE	(	PlacementSpec	,
		ffi::ObjectRef	,
		PlacementSpecNode
	)

---

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

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