tf.nn.rnn_cell.DeviceWrapper

View source on GitHub

Class DeviceWrapper

Operator that ensures an RNNCell runs on a particular device.

Aliases:

• Class tf.compat.v1.nn.rnn_cell.DeviceWrapper
• Class tf.contrib.rnn.DeviceWrapper

__init__

View source

__init__(
    *args,
    **kwargs
)

Construct a DeviceWrapper for cell with device device.

Ensures the wrapped cell is called with tf.device(device).

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.

Properties

graph

DEPRECATED FUNCTION

output_size

scope_name

state_size

Methods

tf.nn.rnn_cell.DeviceWrapper.get_initial_state

View source

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

tf.nn.rnn_cell.DeviceWrapper.zero_state

View source

zero_state(
    batch_size,
    dtype
)