Enum tensorflow_proto::tensorflow::tpu::tpu_embedding_configuration::ShardingStrategy

#[repr(i32)]pub enum ShardingStrategy {
    DivDefault,
    Mod,
}

Sharding strategy of the embedding tables among the hosts. If the sharding_strategy is "mod", each id is assigned to host "id % num_hosts". For instance, 13 ids are split across 5 hosts as: [[0, 5, 10], [1, 6, 11], [2, 7, 12], [3, 8], [4, 9]]. If the sharding_strategy is "div", ids are assigned to hosts in a contiguous manner. In this case, 13 ids are split across 5 hosts as: [[0, 1, 2], [3, 4, 5], [6, 7, 8], [9, 10], [11, 12]]. In both the strategies, if the id space does not evenly divide the number of hosts, each of the first "table_descriptor.vocabulary_size % num_hosts" hosts will be assigned one more id. This partitioning strategy exactly follows that in the embedding_lookup TensorFlow function at tensorflow/python/ops/embedding_ops.py.

Variants

DivDefault

Mod

Implementations

impl ShardingStrategy

pub fn is_valid(value: i32) -> bool

Returns true if value is a variant of ShardingStrategy.

pub fn from_i32(value: i32) -> Option<ShardingStrategy>

Converts an i32 to a ShardingStrategy, or None if value is not a valid variant.

Trait Implementations

impl Clone for ShardingStrategy

pub fn clone(&self) -> ShardingStrategy

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Copy for ShardingStrategy

impl Debug for ShardingStrategy

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

Formats the value using the given formatter.

impl Default for ShardingStrategy

pub fn default() -> ShardingStrategy

Returns the "default value" for a type.

impl Eq for ShardingStrategy

impl Hash for ShardingStrategy

pub fn hash<__H: Hasher>(&self, state: &mut __H)

Feeds this value into the given Hasher.

pub fn hash_slice<H>(data: &[Self], state: &mut H) where
    H: Hasher, 

Feeds a slice of this type into the given Hasher.

impl Ord for ShardingStrategy

pub fn cmp(&self, other: &ShardingStrategy) -> Ordering

This method returns an Ordering between self and other.

#[must_use]pub fn max(self, other: Self) -> Self

Compares and returns the maximum of two values.

#[must_use]pub fn min(self, other: Self) -> Self

Compares and returns the minimum of two values.

#[must_use]pub fn clamp(self, min: Self, max: Self) -> Self

Restrict a value to a certain interval.

impl PartialEq<ShardingStrategy> for ShardingStrategy

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

This method tests for self and other values to be equal, and is used by ==.

#[must_use]pub fn ne(&self, other: &Rhs) -> bool

This method tests for !=.

impl PartialOrd<ShardingStrategy> for ShardingStrategy

pub fn partial_cmp(&self, other: &ShardingStrategy) -> Option<Ordering>

This method returns an ordering between self and other values if one exists.

#[must_use]pub fn lt(&self, other: &Rhs) -> bool

This method tests less than (for self and other) and is used by the < operator.

#[must_use]pub fn le(&self, other: &Rhs) -> bool

This method tests less than or equal to (for self and other) and is used by the <= operator.

#[must_use]pub fn gt(&self, other: &Rhs) -> bool

This method tests greater than (for self and other) and is used by the > operator.

#[must_use]pub fn ge(&self, other: &Rhs) -> bool

This method tests greater than or equal to (for self and other) and is used by the >= operator.

impl StructuralEq for ShardingStrategy

impl StructuralPartialEq for ShardingStrategy