Package org.platanios.tensorflow.proto
Enum RewriterConfig.MemOptType
- java.lang.Object
-
- java.lang.Enum<RewriterConfig.MemOptType>
-
- org.platanios.tensorflow.proto.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 enumorg.platanios.tensorflow.proto.RewriterConfig.MemOptType
-
-
Enum Constant Summary
Enum Constants Enum Constant Description DEFAULT_MEM_OPT
The default setting (SCHEDULING and SWAPPING HEURISTICS only)HEURISTICS
Use any combination of swapping and recomputation heuristics.MANUAL
Driven by manual op-level annotations.NO_MEM_OPT
Disabled in the meta-optimizer.RECOMPUTATION_HEURISTICS
Recomputation heuristics will recompute ops (such as Relu activation) during backprop instead of storing them, reducing peak memory usage.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.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.UNRECOGNIZED
-
Field Summary
Fields Modifier and Type Field Description static int
DEFAULT_MEM_OPT_VALUE
The default setting (SCHEDULING and SWAPPING HEURISTICS only)static int
HEURISTICS_VALUE
Use any combination of swapping and recomputation heuristics.static int
MANUAL_VALUE
Driven by manual op-level annotations.static int
NO_MEM_OPT_VALUE
Disabled in the meta-optimizer.static int
RECOMPUTATION_HEURISTICS_VALUE
Recomputation heuristics will recompute ops (such as Relu activation) during backprop instead of storing them, reducing peak memory usage.static 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.static 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.
-
Method Summary
All Methods Static Methods Instance Methods Concrete Methods Deprecated Methods Modifier and Type Method Description static RewriterConfig.MemOptType
forNumber(int value)
static com.google.protobuf.Descriptors.EnumDescriptor
getDescriptor()
com.google.protobuf.Descriptors.EnumDescriptor
getDescriptorForType()
int
getNumber()
com.google.protobuf.Descriptors.EnumValueDescriptor
getValueDescriptor()
static com.google.protobuf.Internal.EnumLiteMap<RewriterConfig.MemOptType>
internalGetValueMap()
static RewriterConfig.MemOptType
valueOf(int value)
Deprecated.static RewriterConfig.MemOptType
valueOf(com.google.protobuf.Descriptors.EnumValueDescriptor desc)
Returns the enum constant of this type with the specified name.static RewriterConfig.MemOptType
valueOf(java.lang.String name)
Returns the enum constant of this type with the specified name.static RewriterConfig.MemOptType[]
values()
Returns an array containing the constants of this enum type, in the order they are declared.
-
-
-
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;
-
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 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 namejava.lang.NullPointerException
- if the argument is null
-
getNumber
public final int getNumber()
- Specified by:
getNumber
in interfacecom.google.protobuf.Internal.EnumLite
- Specified by:
getNumber
in interfacecom.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 namejava.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 interfacecom.google.protobuf.ProtocolMessageEnum
-
getDescriptorForType
public final com.google.protobuf.Descriptors.EnumDescriptor getDescriptorForType()
- Specified by:
getDescriptorForType
in interfacecom.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 namejava.lang.NullPointerException
- if the argument is null
-
-