tvm::relax::distributed::Axis Struct Reference

tensor axis More...

#include <axis_group_graph.h>

Collaboration diagram for tvm::relax::distributed::Axis:

Public Member Functions
	Axis (const ExprNode *tensor, int dim, int tuple_index=0)
bool	operator== (const Axis &other) const

Public Attributes
const ExprNode *	tensor
int	dim = 0
int	tuple_index = 0

Detailed Description

tensor axis

Constructor & Destructor Documentation

Axis()

tvm::relax::distributed::Axis::Axis ( const ExprNode *  tensor, int  dim, int  tuple_index = 0  )

inline

Member Function Documentation

operator==()

bool tvm::relax::distributed::Axis::operator== ( const Axis &  other ) const

inline

Member Data Documentation

dim

int tvm::relax::distributed::Axis::dim = 0

tensor

const ExprNode* tvm::relax::distributed::Axis::tensor

tuple_index

int tvm::relax::distributed::Axis::tuple_index = 0

---

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

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