tf.nn.RNNCellDeviceWrapper

View source on GitHub

Operator that ensures an RNNCell runs on a particular device.

View aliases

Compat aliases for migration

See Migration guide for more details.

tf.compat.v2.nn.RNNCellDeviceWrapper

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
)