ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tf.nn.RNNCellDeviceWrapper

View source on GitHub

Operator that ensures an RNNCell runs on a particular device.

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

Methods

get_initial_state

View source

zero_state

View source