Enum ConvolutionLayer.AlgoMode
- java.lang.Object
-
- java.lang.Enum<ConvolutionLayer.AlgoMode>
-
- org.deeplearning4j.nn.conf.layers.ConvolutionLayer.AlgoMode
-
- All Implemented Interfaces:
Serializable
,Comparable<ConvolutionLayer.AlgoMode>
- Enclosing class:
- ConvolutionLayer
public static enum ConvolutionLayer.AlgoMode extends Enum<ConvolutionLayer.AlgoMode>
The "PREFER_FASTEST" mode will pick the fastest algorithm for the specified parameters from theConvolutionLayer.FwdAlgo
,ConvolutionLayer.BwdFilterAlgo
, andConvolutionLayer.BwdDataAlgo
lists, but they may be very memory intensive, so if weird errors occur when using cuDNN, please try the "NO_WORKSPACE" mode. Alternatively, it is possible to specify the algorithm manually by setting the "USER_SPECIFIED" mode, but this is not recommended.Note: Currently only supported with cuDNN.
-
-
Enum Constant Summary
Enum Constants Enum Constant Description NO_WORKSPACE
PREFER_FASTEST
USER_SPECIFIED
-
Method Summary
All Methods Static Methods Concrete Methods Modifier and Type Method Description static ConvolutionLayer.AlgoMode
valueOf(String name)
Returns the enum constant of this type with the specified name.static ConvolutionLayer.AlgoMode[]
values()
Returns an array containing the constants of this enum type, in the order they are declared.
-
-
-
Enum Constant Detail
-
NO_WORKSPACE
public static final ConvolutionLayer.AlgoMode NO_WORKSPACE
-
PREFER_FASTEST
public static final ConvolutionLayer.AlgoMode PREFER_FASTEST
-
USER_SPECIFIED
public static final ConvolutionLayer.AlgoMode USER_SPECIFIED
-
-
Method Detail
-
values
public static ConvolutionLayer.AlgoMode[] 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 (ConvolutionLayer.AlgoMode c : ConvolutionLayer.AlgoMode.values()) System.out.println(c);
- Returns:
- an array containing the constants of this enum type, in the order they are declared
-
valueOf
public static ConvolutionLayer.AlgoMode 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 nameNullPointerException
- if the argument is null
-
-