Skip to content

GitLab

".github/vscode:/vscode.git/clone" did not exist on "bad7c26fdc7fb87099dc833ebf8ff873cef5170b"
Commit 4c5a5012 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 7fdde945
Show whitespace changes
Inline Side-by-side
Showing with 2 additions and 0 deletions
official/nlp/modeling/ops/__init__.py
View file @ 4c5a5012
......@@ -14,5 +14,7 @@
"""Ops package definition."""
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_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