Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "3e8c263c818b33afbc1fb9240f67d78361c7355f"
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
Hide 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