diff --git a/python/paddle/nn/__init__.py b/python/paddle/nn/__init__.py index 82fec5c0faa2efccdb036464299e129cef4fb5b2..427a2f91c963f616d923abef44e48da71ff43c94 100644 --- a/python/paddle/nn/__init__.py +++ b/python/paddle/nn/__init__.py @@ -44,8 +44,8 @@ from .control_flow import while_loop #DEFINE_ALIAS # from .control_flow import rnn #DEFINE_ALIAS # from .decode import BeamSearchDecoder #DEFINE_ALIAS # from .decode import Decoder #DEFINE_ALIAS -from .decode import beam_search #DEFINE_ALIAS -from .decode import beam_search_decode #DEFINE_ALIAS +# from .decode import beam_search #DEFINE_ALIAS +# from .decode import beam_search_decode #DEFINE_ALIAS # from .decode import crf_decoding #DEFINE_ALIAS # from .decode import ctc_greedy_decoder #DEFINE_ALIAS # from .decode import dynamic_decode #DEFINE_ALIAS diff --git a/python/paddle/nn/decode.py b/python/paddle/nn/decode.py index f01a5ed15b650ca41ea998ec35fee2c40a24b610..214744217e957f2ea2a800e184d15e7856aecc41 100644 --- a/python/paddle/nn/decode.py +++ b/python/paddle/nn/decode.py @@ -13,16 +13,16 @@ # limitations under the License. # TODO: define api to implement decoding algorithm -from ..fluid.layers import beam_search #DEFINE_ALIAS -from ..fluid.layers import beam_search_decode #DEFINE_ALIAS +# from ..fluid.layers import beam_search #DEFINE_ALIAS +# from ..fluid.layers import beam_search_decode #DEFINE_ALIAS from ..fluid.layers import gather_tree #DEFINE_ALIAS __all__ = [ # 'BeamSearchDecoder', # 'Decoder', - 'beam_search', - 'beam_search_decode', + # 'beam_search', + # 'beam_search_decode', # 'crf_decoding', # 'ctc_greedy_decoder', # 'dynamic_decode',