Module: dragnn_ops

Groups the DRAGNN TensorFlow ops in one module.

Members

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