#include <struct_info.h>
Inheritance diagram for tvm::relax::distributed::ShardingNode:
Collaboration diagram for tvm::relax::distributed::ShardingNode:
Public Member Functions | |
| TVM_FFI_DECLARE_OBJECT_INFO_FINAL ("relax.distributed.Sharding", ShardingNode, PlacementSpecNode) | |
 Public Member Functions inherited from tvm::relax::distributed::PlacementSpecNode | |
| TVM_FFI_DECLARE_OBJECT_INFO ("relax.distributed.PlacementSpec", 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 TVMFFISEqHashKind | _type_s_eq_hash_kind = kTVMFFISEqHashKindConstTreeNode |
Member Function Documentation
◆ RegisterReflection()
|
inlinestatic |
◆ TVM_FFI_DECLARE_OBJECT_INFO_FINAL()
| tvm::relax::distributed::ShardingNode::TVM_FFI_DECLARE_OBJECT_INFO_FINAL | ( | "relax.distributed.Sharding" | , |
| 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
