tf.nn.RNNCellDeviceWrapper
Operator that ensures an RNNCell runs on a particular device.
Inherits From: AbstractRNNCell
, Layer
, Module
tf.nn.RNNCellDeviceWrapper( *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 | |
---|---|
output_size |
|
state_size |
Methods
get_initial_state
get_initial_state( inputs=None, batch_size=None, dtype=None )
zero_state
zero_state( batch_size, dtype )
© 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