#include <struct_info.h>
Inheritance diagram for tvm::relax::distributed::ShardingNode:
Collaboration diagram for tvm::relax::distributed::ShardingNode:
Public Member Functions | |
| TVM_DECLARE_FINAL_OBJECT_INFO (ShardingNode, PlacementSpecNode) | |
 Public Member Functions inherited from tvm::relax::distributed::PlacementSpecNode | |
| TVM_DECLARE_BASE_OBJECT_INFO (PlacementSpecNode, Object) | |
Static Public Member Functions | |
| static void | RegisterReflection () |
| Static Public Member Functions inherited from tvm::relax::distributed::PlacementSpecNode | |
| static void | RegisterReflection () |
Public Attributes | |
| Integer | sharding_dim |
| The dimension of tensor we shard. More... | |
| Public Attributes inherited from tvm::relax::distributed::PlacementSpecNode | |
| 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... | |
Additional Inherited Members | |
| Static Public Attributes inherited from tvm::relax::distributed::PlacementSpecNode | |
| static constexpr const char * | _type_key = "relax.distributed.PlacementSpec" |
| static constexpr TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode |
Member Function Documentation
◆ RegisterReflection()
|
inlinestatic |
◆ TVM_DECLARE_FINAL_OBJECT_INFO()
| tvm::relax::distributed::ShardingNode::TVM_DECLARE_FINAL_OBJECT_INFO | ( | ShardingNode | , |
| PlacementSpecNode | |||
| ) |
Member Data Documentation
◆ sharding_dim
| Integer tvm::relax::distributed::ShardingNode::sharding_dim |
The dimension of tensor we shard.
The documentation for this class was generated from the following file:
- include/tvm/relax/distributed/struct_info.h
