提交 23ac769f 编写于 作者: H Hongkun Yu 提交者: A. Unique TensorFlower

export SequenceBeamSearch and SamplingModule to ops package.

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

PiperOrigin-RevId: 448816131
上级 a77f240f
......@@ -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% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册