tf.contrib.cudnn_rnn.CudnnGRUSaveable
SaveableObject implementation handling Cudnn GRU opaque params.
tf.contrib.cudnn_rnn.CudnnGRUSaveable(
opaque_params, num_layers, num_units, input_size,
input_mode=CUDNN_INPUT_LINEAR_MODE, direction=CUDNN_RNN_UNIDIRECTION,
scope=None, name='cudnn_rnn_saveable'
)
Args |
opaque_params |
a variable, Cudnn RNN opaque params. |
num_layers |
the number of layers for the RNN model. |
num_units |
the number of units within the RNN model. |
input_size |
the size of the input, it could be different from the num_units. |
input_mode |
indicate whether there is a linear projection between the input and the actual computation before the first layer. It could be 'linear_input', 'skip_input' or 'auto_select'. 'linear_input' (default) always applies a linear projection of input onto RNN hidden state. (standard RNN behavior). 'skip_input' is only allowed when input_size == num_units; 'auto_select' implies 'skip_input' when input_size == num_units; otherwise, it implies 'linear_input'. |
direction |
the direction model that the model operates. Could be either 'unidirectional' or 'bidirectional' |
scope |
string of VariableScope, the scope of equivalent subgraph consisting only platform-independent tf RNN cells. |
name |
the name of the CudnnOpaqueParamsSaveable object. |
Attributes |
device |
The device for SaveSpec Tensors. |
format_converter |
|
optional_restore |
A hint to restore assertions that this object is optional. |
Methods
restore
View source
restore(
restored_tensors, restored_shapes
)
Restores this object from 'restored_tensors'.
Args |
restored_tensors |
the tensors that were loaded from a checkpoint |
restored_shapes |
the shapes this object should conform to after restore, or None. |
Returns |
An operation that restores the state of the object. |
Raises |
ValueError |
If the object cannot be restored using the provided parameters. |