ShardingSpecProto
data class ShardingSpecProto(val tensorName: String = "", val device: List<Long> = emptyList(), val indexToDeviceGroupMap: List<IntIntListEntryProto> = emptyList(), val shardedDim: List<ShardedDimProto> = emptyList(), val unknownFields: Map<Int, UnknownField> = emptyMap()) : Message(source)
Constructors
Link copied to clipboard
constructor(tensorName: String = "", device: List<Long> = emptyList(), indexToDeviceGroupMap: List<IntIntListEntryProto> = emptyList(), shardedDim: List<ShardedDimProto> = emptyList(), unknownFields: Map<Int, UnknownField> = emptyMap())