tf.compat.v1.nn.rnn_cell.DeviceWrapper

View source on GitHub

Operator that ensures an RNNCell runs on a particular device.

tf.compat.v1.nn.rnn_cell.DeviceWrapper(
    *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:

• graph: DEPRECATED FUNCTION

  Warning: THIS FUNCTION IS DEPRECATED. It will be removed in a future version. Instructions for updating: Stop using this property because tf.layers layers no longer track their graph.

• output_size

• scope_name

• state_size

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
)