org.tensorflow.framework

Enum RewriterConfig.MemOptType

java.lang.Object

java.lang.Enum<RewriterConfig.MemOptType>

org.tensorflow.framework.RewriterConfig.MemOptType

All Implemented Interfaces:

Serializable, Comparable<RewriterConfig.MemOptType>, org.nd4j.shade.protobuf.Internal.EnumLite, org.nd4j.shade.protobuf.ProtocolMessageEnum

Enclosing class:

RewriterConfig

public static enum RewriterConfig.MemOptType
extends Enum<RewriterConfig.MemOptType>
implements org.nd4j.shade.protobuf.ProtocolMessageEnum

Protobuf enum tensorflow.RewriterConfig.MemOptType

Enum Constant Summary

Enum Constants

Enum Constant and 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 and 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

Modifier and Type	Method and Description
static RewriterConfig.MemOptType	forNumber(int value)
static org.nd4j.shade.protobuf.Descriptors.EnumDescriptor	getDescriptor()
org.nd4j.shade.protobuf.Descriptors.EnumDescriptor	getDescriptorForType()
int	getNumber()
org.nd4j.shade.protobuf.Descriptors.EnumValueDescriptor	getValueDescriptor()
static org.nd4j.shade.protobuf.Internal.EnumLiteMap<RewriterConfig.MemOptType>	internalGetValueMap()
static RewriterConfig.MemOptType	valueOf(org.nd4j.shade.protobuf.Descriptors.EnumValueDescriptor desc)
static RewriterConfig.MemOptType	valueOf(int value) Deprecated. Use forNumber(int) instead.
static RewriterConfig.MemOptType	valueOf(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.

Methods inherited from class java.lang.Enum

clone, compareTo, equals, finalize, getDeclaringClass, hashCode, name, ordinal, toString, valueOf

Methods inherited from class java.lang.Object

getClass, notify, notifyAll, wait, wait, wait

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(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:

IllegalArgumentException - if this enum type has no constant with the specified name

NullPointerException - if the argument is null

getNumber

public final int getNumber()

Specified by:

getNumber in interface org.nd4j.shade.protobuf.Internal.EnumLite

Specified by:

getNumber in interface org.nd4j.shade.protobuf.ProtocolMessageEnum

valueOf

@Deprecated
public static RewriterConfig.MemOptType valueOf(int value)

Deprecated. Use forNumber(int) instead.

forNumber

public static RewriterConfig.MemOptType forNumber(int value)

internalGetValueMap

public static org.nd4j.shade.protobuf.Internal.EnumLiteMap<RewriterConfig.MemOptType> internalGetValueMap()

getValueDescriptor

public final org.nd4j.shade.protobuf.Descriptors.EnumValueDescriptor getValueDescriptor()

Specified by:

getValueDescriptor in interface org.nd4j.shade.protobuf.ProtocolMessageEnum

getDescriptorForType

public final org.nd4j.shade.protobuf.Descriptors.EnumDescriptor getDescriptorForType()

Specified by:

getDescriptorForType in interface org.nd4j.shade.protobuf.ProtocolMessageEnum

getDescriptor

public static final org.nd4j.shade.protobuf.Descriptors.EnumDescriptor getDescriptor()

valueOf

public static RewriterConfig.MemOptType valueOf(org.nd4j.shade.protobuf.Descriptors.EnumValueDescriptor desc)