Enum tensorflow_proto::tensorflow::rewriter_config::MemOptType

#[repr(i32)]pub enum MemOptType {
    DefaultMemOpt,
    NoMemOpt,
    Manual,
    SwappingHeuristics,
    RecomputationHeuristics,
    SchedulingHeuristics,
    Heuristics,
}

Variants

DefaultMemOpt

The default setting (SCHEDULING and SWAPPING HEURISTICS only)

NoMemOpt

Disabled in the meta-optimizer.

Manual

Driven by manual op-level annotations.

SwappingHeuristics

Swapping heuristic will move a tensor from the GPU to the CPU and move it back when needed to reduce peak memory usage.

RecomputationHeuristics

Recomputation heuristics will recompute ops (such as Relu activation) during backprop instead of storing them, reducing peak memory usage.

SchedulingHeuristics

Scheduling will split big ops such as AddN and try to enforce a schedule of the new computations that decreases peak memory usage.

Heuristics

Use any combination of swapping and recomputation heuristics.

Implementations

impl MemOptType

pub fn is_valid(value: i32) -> bool

Returns true if value is a variant of MemOptType.

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

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

Trait Implementations

impl Clone for MemOptType

pub fn clone(&self) -> MemOptType

Returns a copy of the value.

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

Performs copy-assignment from source.

impl Copy for MemOptType

impl Debug for MemOptType

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

Formats the value using the given formatter.

impl Default for MemOptType

pub fn default() -> MemOptType

Returns the "default value" for a type.

impl Eq for MemOptType

impl Hash for MemOptType

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 MemOptType

pub fn cmp(&self, other: &MemOptType) -> 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<MemOptType> for MemOptType

pub fn eq(&self, other: &MemOptType) -> 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<MemOptType> for MemOptType

pub fn partial_cmp(&self, other: &MemOptType) -> 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 MemOptType

impl StructuralPartialEq for MemOptType