Package org.deeplearning4j.nn.conf.layers

Enum ConvolutionLayer.FwdAlgo

java.lang.Object

java.lang.Enum<ConvolutionLayer.FwdAlgo>

org.deeplearning4j.nn.conf.layers.ConvolutionLayer.FwdAlgo

All Implemented Interfaces:

Serializable, Comparable<ConvolutionLayer.FwdAlgo>

Enclosing class:

ConvolutionLayer

public static enum ConvolutionLayer.FwdAlgo
extends Enum<ConvolutionLayer.FwdAlgo>

The forward algorithm to use when ConvolutionLayer.AlgoMode is set to "USER_SPECIFIED".

Note: Currently only supported with cuDNN.

Enum Constant Summary

Enum Constants

Enum Constant	Description
COUNT
DIRECT
FFT
FFT_TILING
GEMM
IMPLICIT_GEMM
IMPLICIT_PRECOMP_GEMM
WINOGRAD
WINOGRAD_NONFUSED

Method Summary

Modifier and Type	Method	Description
static ConvolutionLayer.FwdAlgo	valueOf(String name)	Returns the enum constant of this type with the specified name.
static ConvolutionLayer.FwdAlgo[]	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

IMPLICIT_GEMM

public static final ConvolutionLayer.FwdAlgo IMPLICIT_GEMM

IMPLICIT_PRECOMP_GEMM

public static final ConvolutionLayer.FwdAlgo IMPLICIT_PRECOMP_GEMM

GEMM

public static final ConvolutionLayer.FwdAlgo GEMM

DIRECT

public static final ConvolutionLayer.FwdAlgo DIRECT

FFT

public static final ConvolutionLayer.FwdAlgo FFT

FFT_TILING

public static final ConvolutionLayer.FwdAlgo FFT_TILING

WINOGRAD

public static final ConvolutionLayer.FwdAlgo WINOGRAD

WINOGRAD_NONFUSED

public static final ConvolutionLayer.FwdAlgo WINOGRAD_NONFUSED

COUNT

public static final ConvolutionLayer.FwdAlgo COUNT

Method Detail

values

public static ConvolutionLayer.FwdAlgo[] 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.FwdAlgo c : ConvolutionLayer.FwdAlgo.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.FwdAlgo 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