public class cudnnRNNMode extends Object
Modifier and Type | Field and Description |
---|---|
static int |
CUDNN_GRU
Using h' = tanh(r * Uh(t-1) + Wx) and h = (1 - z) * h' + z * h(t-1);
|
static int |
CUDNN_LSTM
LSTM with no peephole connections
|
static int |
CUDNN_RNN_RELU
basic RNN cell type with ReLu activation
|
static int |
CUDNN_RNN_TANH
basic RNN cell type with tanh activation
|
Modifier and Type | Method and Description |
---|---|
static String |
stringFor(int n)
Returns a string representation of the given constant
|
public static final int CUDNN_RNN_RELU
public static final int CUDNN_RNN_TANH
public static final int CUDNN_LSTM
public static final int CUDNN_GRU
public static String stringFor(int n)
Copyright © 2020. All rights reserved.