tf.contrib.rnn.CompiledWrapper

Wraps step execution in an XLA JIT scope.

Inherits From: RNNCell

tf.contrib.rnn.CompiledWrapper(
    cell, compile_stateful=False
)

Args
cell	Instance of RNNCell.
compile_stateful	Whether to compile stateful ops like initializers and random number generators (default: False).

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	Integer or TensorShape: size of outputs produced by this cell.
scope_name
state_size	size(s) of state(s) used by this cell. It can be represented by an Integer, a TensorShape or a tuple of Integers or TensorShapes.

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
)

Return zero-filled state tensor(s).

Args
batch_size	int, float, or unit Tensor representing the batch size.
dtype	the data type to use for the state.

Returns
If state_size is an int or TensorShape, then the return value is a N-D tensor of shape [batch_size, state_size] filled with zeros. If state_size is a nested list or tuple, then the return value is a nested list or tuple (of the same structure) of 2-D tensors with the shapes [batch_size, s] for each s in state_size.