code: def __init__(
      self,
      num_chars,
      num_speakers,
      r,
      postnet_output_dim=80,
      decoder_output_dim=80,
      attn_type="original",
      attn_win=False,
      attn_norm="softmax",
      attn_K=4,
      prenet_type="original",
      prenet_dropout=True,
      forward_attn=False,
      trans_agent=False,
      forward_attn_mask=False,
      location_attn=True,
      separate_stopnet=True,
      bidirectional_decoder=False,
      enable_tflite=False,
  ):
      super(Tacotron2, self).__init__()
      self.r = r
      self.decoder_output_dim = decoder_output_dim
      self.postnet_output_dim = postnet_output_dim
      self.bidirectional_decoder = bidirectional_decoder
      self.num_speakers = num_speakers
      self.speaker_embed_dim = 256
      self.enable_tflite = enable_tflite
  
      self.embedding = keras.layers.Embedding(
          num_chars, 512, name="embedding"
      )
      self.encoder = Encoder(512, name="encoder")
      # TODO: most of the decoder args have no use at the momment
      self.decoder = Decoder(
          decoder_output_dim,
          r,
          attn_type=attn_type,
          use_attn_win=attn_win,
          attn_norm=attn_norm,
          prenet_type=prenet_type,
          prenet_dropout=prenet_dropout,
          use_forward_attn=forward_attn,
          use_trans_agent=trans_agent,
          use_forward_attn_mask=forward_attn_mask,
          use_location_attn=location_attn,
          attn_K=attn_K,
          separate_stopnet=separate_stopnet,
          speaker_emb_dim=self.speaker_embed_dim,
          name="decoder",
          enable_tflite=enable_tflite,
      )
      self.postnet = Postnet(postnet_output_dim, 5, name="postnet")
  
file_path: TTS/TTS/tts/tf/models/tacotron2.py
imports: import tensorflow
  from tensorflow import keras
  from TTS.tts.tf.layers.tacotron2 import Encoder, Decoder, Postnet
  from TTS.tts.tf.utils.tf_utils import shape_list
label_desc: Use image node_id 1 to create a new Tacotron2 object from inherited base classes: keras.models with example: obj = Tacotron2(num_chars, num_speakers, r, postnet_output_dim, decoder_output_dim, attn_type, attn_win, attn_norm, attn_K, prenet_type, prenet_dropout, forward_attn, trans_agent, forward_attn_mask, location_attn, separate_stopnet, bidirectional_decoder, enable_tflite)