Defined in
tensorflow/dragnn/python/dragnn_ops.py.
Groups the DRAGNN TensorFlow ops in one module.
advance_from_oracle(...): Given a
handle to a ComputeSession, advances based on the next oracle action.
advance_from_prediction(...): Given
a handle to a ComputeSession and a tensor of scores, advances the state.
attach_data_reader(...): Given a handle
to a ComputeSession, attaches a data source.
batch_size(...): Given a handle to a
ComputeSession, returns the batch size of the given component.
bulk_advance_from_oracle(...):
Given a handle to a ComputeSession, advances until all states are final. Note
bulk_advance_from_prediction(...):
Given a handle to a ComputeSession and a tensor of scores, advances the state
until
bulk_fixed_embeddings(...): This op
is a more efficient version of BulkFixedFeatures to be run with large
bulk_fixed_feature_ids(...): This op
is a variant of BulkFixedFeatures that only outputs fixed feature IDs.
bulk_fixed_features(...): Given a
handle to a ComputeSession and a component name, outputs fixed features
dragnn_embedding_initializer(...):
Reads embeddings from an sstable of dist_belief.TokenEmbedding protos for
emit_all_final(...): Given a handle to a
ComputeSession, emits a single bool indicating whether all
emit_annotations(...): Given a handle to a
ComputeSession, emits a vector of strings corresponding to the
emit_oracle_labels(...): Given a handle
to a ComputeSession, emits a vector of gold labels.
extract_fixed_features(...): Given a
handle to a ComputeSession and a channel index, outputs fixed features.
extract_link_features(...): Given a
handle to a ComputeSession and a channel index, outputs link features.
get_component_trace(...): Gets the raw
MasterTrace proto for each batch, state, and beam slot.
get_session(...): Given MasterSpec and
GridPoint protos, outputs a handle to a ComputeSession.
init_component_data(...): Given a
handle to a ComputeSession, initializes the given component with the
release_session(...): Given a handle to the
ComputeSession, deletes it from memory.
set_tracing(...): Given a handle to a
ComputeSession, turns on or off tracing.
write_annotations(...): Given a handle to
a ComputeSession, has the given component write its