tvm
Public Member Functions | Static Public Member Functions | List of all members
tvm::relax::distributed::PlacementSpec Class Reference

Managed reference to PlacementSpecNode. More...

#include <struct_info.h>

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

Public Member Functions

 TVM_DEFINE_OBJECT_REF_METHODS (PlacementSpec, ObjectRef, PlacementSpecNode)
 

Static Public Member Functions

static PlacementSpec Sharding (int axis)
 
static PlacementSpec Replica ()
 

Detailed Description

Managed reference to PlacementSpecNode.

See also
PlacementSpecNode

Member Function Documentation

◆ Replica()

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

◆ Sharding()

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

◆ TVM_DEFINE_OBJECT_REF_METHODS()

tvm::relax::distributed::PlacementSpec::TVM_DEFINE_OBJECT_REF_METHODS ( PlacementSpec  ,
ObjectRef  ,
PlacementSpecNode   
)

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