View source on GitHub |
Operator that ensures an RNNCell runs on a particular device.
Inherits From: AbstractRNNCell
, Layer
, Module
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
)