Coverage for /pythoncovmergedfiles/medio/medio/usr/local/lib/python3.8/site-packages/tensorflow_addons/seq2seq/__init__.py: 100%
33 statements
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
« prev ^ index » next coverage.py v7.4.0, created at 2024-01-03 07:57 +0000
1# Copyright 2019 The TensorFlow Authors. All Rights Reserved.
2#
3# Licensed under the Apache License, Version 2.0 (the "License");
4# you may not use this file except in compliance with the License.
5# You may obtain a copy of the License at
6#
7# http://www.apache.org/licenses/LICENSE-2.0
8#
9# Unless required by applicable law or agreed to in writing, software
10# distributed under the License is distributed on an "AS IS" BASIS,
11# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
12# See the License for the specific language governing permissions and
13# limitations under the License.
14# ==============================================================================
15"""Additional layers for sequence to sequence models."""
17from tensorflow_addons.seq2seq.attention_wrapper import AttentionMechanism
18from tensorflow_addons.seq2seq.attention_wrapper import AttentionWrapper
19from tensorflow_addons.seq2seq.attention_wrapper import AttentionWrapperState
20from tensorflow_addons.seq2seq.attention_wrapper import BahdanauAttention
21from tensorflow_addons.seq2seq.attention_wrapper import BahdanauMonotonicAttention
22from tensorflow_addons.seq2seq.attention_wrapper import LuongAttention
23from tensorflow_addons.seq2seq.attention_wrapper import LuongMonotonicAttention
24from tensorflow_addons.seq2seq.attention_wrapper import hardmax
25from tensorflow_addons.seq2seq.attention_wrapper import monotonic_attention
26from tensorflow_addons.seq2seq.attention_wrapper import safe_cumprod
28from tensorflow_addons.seq2seq.basic_decoder import BasicDecoder
29from tensorflow_addons.seq2seq.basic_decoder import BasicDecoderOutput
31from tensorflow_addons.seq2seq.beam_search_decoder import BeamSearchDecoder
32from tensorflow_addons.seq2seq.beam_search_decoder import BeamSearchDecoderOutput
33from tensorflow_addons.seq2seq.beam_search_decoder import BeamSearchDecoderState
34from tensorflow_addons.seq2seq.beam_search_decoder import FinalBeamSearchDecoderOutput
35from tensorflow_addons.seq2seq.beam_search_decoder import gather_tree
36from tensorflow_addons.seq2seq.beam_search_decoder import gather_tree_from_array
37from tensorflow_addons.seq2seq.beam_search_decoder import tile_batch
39from tensorflow_addons.seq2seq.decoder import BaseDecoder
40from tensorflow_addons.seq2seq.decoder import Decoder
41from tensorflow_addons.seq2seq.decoder import dynamic_decode
43from tensorflow_addons.seq2seq.loss import SequenceLoss
44from tensorflow_addons.seq2seq.loss import sequence_loss
46from tensorflow_addons.seq2seq.sampler import CustomSampler
47from tensorflow_addons.seq2seq.sampler import GreedyEmbeddingSampler
48from tensorflow_addons.seq2seq.sampler import InferenceSampler
49from tensorflow_addons.seq2seq.sampler import SampleEmbeddingSampler
50from tensorflow_addons.seq2seq.sampler import Sampler
51from tensorflow_addons.seq2seq.sampler import ScheduledEmbeddingTrainingSampler
52from tensorflow_addons.seq2seq.sampler import ScheduledOutputTrainingSampler
53from tensorflow_addons.seq2seq.sampler import TrainingSampler