View source on GitHub
|
Operator that ensures an RNNCell runs on a particular device.
tf.nn.RNNCellDeviceWrapper(
*args, **kwargs
)
Args | |
|---|---|
cell
|
An instance of RNNCell.
|
device
|
A device string or function, for passing to tf.device.
|
**kwargs
|
dict of keyword arguments for base layer. |
Attributes | |
|---|---|
output_size
|
|
state_size
|
|
Methods
get_initial_state
get_initial_state(
inputs=None, batch_size=None, dtype=None
)
zero_state
zero_state(
batch_size, dtype
)