TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
model.Tacotron2 Class Reference
Inheritance diagram for model.Tacotron2:
Collaboration diagram for model.Tacotron2:

Public Member Functions

def __init__ (self, mask_padding, n_mel_channels, n_symbols, symbols_embedding_dim, encoder_kernel_size, encoder_n_convolutions, encoder_embedding_dim, attention_rnn_dim, attention_dim, attention_location_n_filters, attention_location_kernel_size, n_frames_per_step, decoder_rnn_dim, prenet_dim, max_decoder_steps, gate_threshold, p_attention_dropout, p_decoder_dropout, postnet_embedding_dim, postnet_kernel_size, postnet_n_convolutions, decoder_no_early_stopping)
 
def parse_batch (self, batch)
 
def parse_output (self, outputs, output_lengths)
 
def forward (self, inputs)
 
def infer (self, inputs, input_lengths)
 

Public Attributes

 mask_padding
 
 n_mel_channels
 
 n_frames_per_step
 
 embedding
 
 encoder
 
 decoder
 
 postnet
 

Constructor & Destructor Documentation

◆ __init__()

def model.Tacotron2.__init__ (   self,
  mask_padding,
  n_mel_channels,
  n_symbols,
  symbols_embedding_dim,
  encoder_kernel_size,
  encoder_n_convolutions,
  encoder_embedding_dim,
  attention_rnn_dim,
  attention_dim,
  attention_location_n_filters,
  attention_location_kernel_size,
  n_frames_per_step,
  decoder_rnn_dim,
  prenet_dim,
  max_decoder_steps,
  gate_threshold,
  p_attention_dropout,
  p_decoder_dropout,
  postnet_embedding_dim,
  postnet_kernel_size,
  postnet_n_convolutions,
  decoder_no_early_stopping 
)

Member Function Documentation

◆ parse_batch()

def model.Tacotron2.parse_batch (   self,
  batch 
)

◆ parse_output()

def model.Tacotron2.parse_output (   self,
  outputs,
  output_lengths 
)
Here is the caller graph for this function:

◆ forward()

def model.Tacotron2.forward (   self,
  inputs 
)
Here is the call graph for this function:

◆ infer()

def model.Tacotron2.infer (   self,
  inputs,
  input_lengths 
)

Member Data Documentation

◆ mask_padding

model.Tacotron2.mask_padding

◆ n_mel_channels

model.Tacotron2.n_mel_channels

◆ n_frames_per_step

model.Tacotron2.n_frames_per_step

◆ embedding

model.Tacotron2.embedding

◆ encoder

model.Tacotron2.encoder

◆ decoder

model.Tacotron2.decoder

◆ postnet

model.Tacotron2.postnet

The documentation for this class was generated from the following file: