Skip to content

GitLab

Commit 23ac769f authored by Hongkun Yu's avatar Hongkun Yu Committed by A. Unique TensorFlower
Browse files

export SequenceBeamSearch and SamplingModule to ops package. 

https://github.com/tensorflow/models/issues/10638

PiperOrigin-RevId: 448816131
parent a77f240f
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/modeling/ops/__init__.py
View file @ 23ac769f
...@@ -14,5 +14,7 @@ ...@@ -14,5 +14,7 @@
"""Ops package definition.""" """Ops package definition."""
from official.nlp.modeling.ops.beam_search import sequence_beam_search from official.nlp.modeling.ops.beam_search import sequence_beam_search
from official.nlp.modeling.ops.beam_search import SequenceBeamSearch
from official.nlp.modeling.ops.sampling_module import SamplingModule
from official.nlp.modeling.ops.segment_extractor import get_next_sentence_labels from official.nlp.modeling.ops.segment_extractor import get_next_sentence_labels
from official.nlp.modeling.ops.segment_extractor import get_sentence_order_labels from official.nlp.modeling.ops.segment_extractor import get_sentence_order_labels
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment