models/official/nlp/modeling/ops
..
__init__.py
beam_search.py
beam_search_test.py
decoding_module.py
decoding_module_test.py
sampling_module.py
segment_extractor.py
segment_extractor_test.py