Package org.tensorflow.framework

Enum Class RewriterConfig.MemOptType

java.lang.Object
java.lang.Enum<RewriterConfig.MemOptType>
org.tensorflow.framework.RewriterConfig.MemOptType
All Implemented Interfaces:
com.google.protobuf.Internal.EnumLite, com.google.protobuf.ProtocolMessageEnum, Serializable, Comparable<RewriterConfig.MemOptType>, Constable
Enclosing class:
RewriterConfig
public static enum RewriterConfig.MemOptType extends Enum<RewriterConfig.MemOptType> implements com.google.protobuf.ProtocolMessageEnum
Protobuf enum tensorflow.RewriterConfig.MemOptType

  • Enum Constant Details

    • DEFAULT_MEM_OPT

      public static final RewriterConfig.MemOptType DEFAULT_MEM_OPT
       The default setting (SCHEDULING and SWAPPING HEURISTICS only)
      DEFAULT_MEM_OPT = 0;

    • NO_MEM_OPT

      public static final RewriterConfig.MemOptType NO_MEM_OPT
       Disabled in the meta-optimizer.
      NO_MEM_OPT = 1;

    • MANUAL

      public static final RewriterConfig.MemOptType MANUAL
       Driven by manual op-level annotations.
      MANUAL = 2;

    • SWAPPING_HEURISTICS

      public static final RewriterConfig.MemOptType SWAPPING_HEURISTICS
       Swapping heuristic will move a tensor from the GPU to the CPU and move
 it back when needed to reduce peak memory usage.
      SWAPPING_HEURISTICS = 4;

    • RECOMPUTATION_HEURISTICS

      public static final RewriterConfig.MemOptType RECOMPUTATION_HEURISTICS
       Recomputation heuristics will recompute ops (such as Relu activation)
 during backprop instead of storing them, reducing peak memory usage.
      RECOMPUTATION_HEURISTICS = 5;

    • SCHEDULING_HEURISTICS

      public static final RewriterConfig.MemOptType SCHEDULING_HEURISTICS
       Scheduling will split big ops such as AddN and try to enforce a schedule
 of the new computations that decreases peak memory usage.
      SCHEDULING_HEURISTICS = 6;

    • HEURISTICS

      public static final RewriterConfig.MemOptType HEURISTICS
       Use any combination of swapping and recomputation heuristics.
      HEURISTICS = 3;

    • UNRECOGNIZED

      public static final RewriterConfig.MemOptType UNRECOGNIZED

  • Field Details

    • DEFAULT_MEM_OPT_VALUE

      public static final int DEFAULT_MEM_OPT_VALUE
       The default setting (SCHEDULING and SWAPPING HEURISTICS only)
      DEFAULT_MEM_OPT = 0;
      See Also:

    • NO_MEM_OPT_VALUE

      public static final int NO_MEM_OPT_VALUE
       Disabled in the meta-optimizer.
      NO_MEM_OPT = 1;
      See Also:

    • MANUAL_VALUE

      public static final int MANUAL_VALUE
       Driven by manual op-level annotations.
      MANUAL = 2;
      See Also:

    • SWAPPING_HEURISTICS_VALUE

      public static final int SWAPPING_HEURISTICS_VALUE
       Swapping heuristic will move a tensor from the GPU to the CPU and move
 it back when needed to reduce peak memory usage.
      SWAPPING_HEURISTICS = 4;
      See Also:

    • RECOMPUTATION_HEURISTICS_VALUE

      public static final int RECOMPUTATION_HEURISTICS_VALUE
       Recomputation heuristics will recompute ops (such as Relu activation)
 during backprop instead of storing them, reducing peak memory usage.
      RECOMPUTATION_HEURISTICS = 5;
      See Also:

    • SCHEDULING_HEURISTICS_VALUE

      public static final int SCHEDULING_HEURISTICS_VALUE
       Scheduling will split big ops such as AddN and try to enforce a schedule
 of the new computations that decreases peak memory usage.
      SCHEDULING_HEURISTICS = 6;
      See Also:

    • HEURISTICS_VALUE

      public static final int HEURISTICS_VALUE
       Use any combination of swapping and recomputation heuristics.
      HEURISTICS = 3;
      See Also:

  • Method Details

    • values

      public static RewriterConfig.MemOptType[] values()
      Returns an array containing the constants of this enum class, in the order they are declared.
      Returns:
      an array containing the constants of this enum class, in the order they are declared

    • valueOf

      public static RewriterConfig.MemOptType valueOf(String name)
      Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)
      Parameters:
      name - the name of the enum constant to be returned.
      Returns:
      the enum constant with the specified name
      Throws:
      IllegalArgumentException - if this enum class has no constant with the specified name
      NullPointerException - if the argument is null

    • getNumber

      public final int getNumber()
      Specified by:
      getNumber in interface com.google.protobuf.Internal.EnumLite
      Specified by:
      getNumber in interface com.google.protobuf.ProtocolMessageEnum

    • valueOf

      @Deprecated public static RewriterConfig.MemOptType valueOf(int value)
      Deprecated.
      Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)
      Parameters:
      value - the name of the enum constant to be returned.
      Returns:
      the enum constant with the specified name
      Throws:
      IllegalArgumentException - if this enum class has no constant with the specified name
      NullPointerException - if the argument is null

    • forNumber

      public static RewriterConfig.MemOptType forNumber(int value)
      Parameters:
      value - The numeric wire value of the corresponding enum entry.
      Returns:
      The enum associated with the given numeric wire value.

    • internalGetValueMap

      public static com.google.protobuf.Internal.EnumLiteMap<RewriterConfig.MemOptType> internalGetValueMap()

    • getValueDescriptor

      public final com.google.protobuf.Descriptors.EnumValueDescriptor getValueDescriptor()
      Specified by:
      getValueDescriptor in interface com.google.protobuf.ProtocolMessageEnum

    • getDescriptorForType

      public final com.google.protobuf.Descriptors.EnumDescriptor getDescriptorForType()
      Specified by:
      getDescriptorForType in interface com.google.protobuf.ProtocolMessageEnum

    • getDescriptor

      public static final com.google.protobuf.Descriptors.EnumDescriptor getDescriptor()

    • valueOf

      public static RewriterConfig.MemOptType valueOf(com.google.protobuf.Descriptors.EnumValueDescriptor desc)
      Returns the enum constant of this class with the specified name. The string must match exactly an identifier used to declare an enum constant in this class. (Extraneous whitespace characters are not permitted.)
      Parameters:
      desc - the name of the enum constant to be returned.
      Returns:
      the enum constant with the specified name
      Throws:
      IllegalArgumentException - if this enum class has no constant with the specified name
      NullPointerException - if the argument is null