tf.nn.RNNCellDeviceWrapper

View source on GitHub

Operator that ensures an RNNCell runs on a particular device.

tf.nn.RNNCellDeviceWrapper(
    *args, **kwargs
)

Args:

Attributes:

Methods

get_initial_state

View source

get_initial_state(
    inputs=None, batch_size=None, dtype=None
)

zero_state

View source

zero_state(
    batch_size, dtype
)