tf.nn.RNNCellDeviceWrapper

Operator that ensures an RNNCell runs on a particular device.

Inherits From: AbstractRNNCell, Layer, Module

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
output_size
state_size

Methods

get_initial_state

View source

zero_state

View source

© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/nn/RNNCellDeviceWrapper