TensorFlow 2.0 Beta is available Learn more

tf.contrib.rnn.CompiledWrapper

View source on GitHub

Class CompiledWrapper

Wraps step execution in an XLA JIT scope.

Inherits From: RNNCell

__init__

View source

__init__(
    cell,
    compile_stateful=False
)

Create CompiledWrapper cell.

Args:

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

Properties

graph

DEPRECATED FUNCTION

output_size

scope_name

state_size

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
)