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