tvm
tvm::relax::distributed::AxisGroupGraph Member List

This is the complete list of members for tvm::relax::distributed::AxisGroupGraph, including all inherited members.

AddPropagationCutPoint(Axis axis, AxisShardingSpec spec)tvm::relax::distributed::AxisGroupGraphinline
AddSrcShardingPoint(Axis axis, AxisShardingSpec spec)tvm::relax::distributed::AxisGroupGraphinline
AxisGroupGraph()=defaulttvm::relax::distributed::AxisGroupGraph
EdgeType enum nametvm::relax::distributed::AxisGroupGraph
GetAxisShardingSpec(Axis axis)tvm::relax::distributed::AxisGroupGraphinline
JoinAxis(Axis axis1, Axis axis2, EdgeType type)tvm::relax::distributed::AxisGroupGraphinline
PropagateShardingSpec()tvm::relax::distributed::AxisGroupGraphinline