org.deeplearning4j.nn.conf.layers

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 the ConvolutionLayer.FwdAlgo, ConvolutionLayer.BwdFilterAlgo, and ConvolutionLayer.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 and Description
NO_WORKSPACE
PREFER_FASTEST
USER_SPECIFIED

Method Summary

Modifier and Type	Method and 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.

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

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 name

NullPointerException - if the argument is null