Package org.platanios.tensorflow.proto

Enum RewriterConfig.MemOptType

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

    • Enum Constant Detail

      • DEFAULT_MEM_OPT

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

      • 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;

    • Field Detail

      • 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:
        Constant Field Values

      • NO_MEM_OPT_VALUE

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

      • MANUAL_VALUE

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

      • 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:
        Constant Field Values

      • 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:
        Constant Field Values

      • 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:
        Constant Field Values

      • HEURISTICS_VALUE

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

    • Method Detail

      • values

        public static RewriterConfig.MemOptType[] values()
        Returns an array containing the constants of this enum type, in the order they are declared. This method may be used to iterate over the constants as follows: 
        for (RewriterConfig.MemOptType c : RewriterConfig.MemOptType.values())
    System.out.println(c);
        Returns:
        an array containing the constants of this enum type, in the order they are declared

      • valueOf

        public static RewriterConfig.MemOptType valueOf​(java.lang.String name)
        Returns the enum constant of this type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (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:
        java.lang.IllegalArgumentException - if this enum type has no constant with the specified name
        java.lang.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 type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (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:
        java.lang.IllegalArgumentException - if this enum type has no constant with the specified name
        java.lang.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 type with the specified name. The string must match exactly an identifier used to declare an enum constant in this type. (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:
        java.lang.IllegalArgumentException - if this enum type has no constant with the specified name
        java.lang.NullPointerException - if the argument is null