# tf.nn.RNNCellDeviceWrapper

<br />

|---|
| [![](https://www.tensorflow.org/images/GitHub-Mark-32px.png) View source on GitHub](https://github.com/tensorflow/tensorflow/blob/v2.0.0/tensorflow/python/keras/layers/rnn_cell_wrapper_v2.py#L115-L122) |

Operator that ensures an RNNCell runs on a particular device.

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

<br />

<br />

<br />

| ## Args ||
|---|---|
| `cell` | An instance of `RNNCell`. |
| `device` | A device string or function, for passing to [`tf.device`](https://www.tensorflow.org/versions/r2.0/api_docs/python/tf/device). |
| `**kwargs` | dict of keyword arguments for base layer. |

<br />

<br />

<br />

<br />

| ## Attributes ||
|---|---|
| `output_size` | <br /> <br /> |
| `state_size` | <br /> <br /> |

<br />

## Methods

### `get_initial_state`

[View source](https://github.com/tensorflow/tensorflow/blob/v2.0.0/tensorflow/python/keras/layers/recurrent.py#L976-L977)

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

### `zero_state`

[View source](https://github.com/tensorflow/tensorflow/blob/v2.0.0/tensorflow/python/ops/rnn_cell_wrapper_impl.py#L428-L431)

    zero_state(
        batch_size, dtype
    )