Struct ShardingSpecProto 

pub struct ShardingSpecProto {
    pub tensor_name: String,
    pub device: Vec<i64>,
    pub index_to_device_group_map: Vec<IntIntListEntryProto>,
    pub sharded_dim: Vec<ShardedDimProto>,
}

ShardingSpecProto: This describes the sharding spec for a specific input or output tensor of a node.

Fields

tensor_name: String

This field MUST be present for this version of the IR. Identifies the input or output of the node that is being sharded. Required to match a name specified in the node’s input or output list of ValueInfoProtos. It is called logical tensor in subsequent descriptions.

device: Vec<i64>

The following is the list of devices across which the logical tensor is sharded or replicated.

index_to_device_group_map: Vec<IntIntListEntryProto>

Each element v in above field devices may represent either a device or a set of devices (when we want the same shard/tensor to be replicated across a subset of devices), as indicated by the following optional map. If the map contains an entry for v, then v represents a device group, and the map indicates the set of devices in that group.

sharded_dim: Vec<ShardedDimProto>

The following is the sharded-shape of the tensor, consisting of the sharding-spec for each axis of the tensor.

Trait Implementations

impl Clone for ShardingSpecProto

fn clone(&self) -> ShardingSpecProto

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for ShardingSpecProto

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.

impl Default for ShardingSpecProto

fn default() -> ShardingSpecProto

Returns the “default value” for a type.

impl Message for ShardingSpecProto

fn encoded_len(&self) -> usize

Returns the encoded length of the message without a length delimiter.

fn clear(&mut self)

Clears the message, resetting all fields to their default.

fn encode(&self, buf: &mut impl BufMut) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message to a buffer.

fn encode_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message to a newly allocated buffer.

fn encode_length_delimited( &self, buf: &mut impl BufMut, ) -> Result<(), EncodeError>

where Self: Sized,

Encodes the message with a length-delimiter to a buffer.

fn encode_length_delimited_to_vec(&self) -> Vec<u8>

where Self: Sized,

Encodes the message with a length-delimiter to a newly allocated buffer.

fn decode(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes an instance of the message from a buffer.

fn decode_length_delimited(buf: impl Buf) -> Result<Self, DecodeError>

where Self: Default,

Decodes a length-delimited instance of the message from the buffer.

fn merge(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes an instance of the message from a buffer, and merges it into self.

fn merge_length_delimited(&mut self, buf: impl Buf) -> Result<(), DecodeError>

where Self: Sized,

Decodes a length-delimited instance of the message from buffer, and merges it into self.

impl PartialEq for ShardingSpecProto

fn eq(&self, other: &ShardingSpecProto) -> bool

Tests for self and other values to be equal, and is used by ==.

fn ne(&self, other: &Rhs) -> bool

Tests for !=. The default implementation is almost always sufficient, and should not be overridden without very good reason.

impl StructuralPartialEq for ShardingSpecProto