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
export_tacotron2_onnx.DecoderIter Class Reference
Inheritance diagram for export_tacotron2_onnx.DecoderIter:
Collaboration diagram for export_tacotron2_onnx.DecoderIter:

Public Member Functions

def __init__ (self, tacotron2)
 
def decode (self, decoder_input, in_attention_hidden, in_attention_cell, in_decoder_hidden, in_decoder_cell, in_attention_weights, in_attention_weights_cum, in_attention_context, memory, processed_memory, mask)
 
def forward (self, decoder_input, attention_hidden, attention_cell, decoder_hidden, decoder_cell, attention_weights, attention_weights_cum, attention_context, memory, processed_memory, mask)
 

Public Attributes

 tacotron2
 
 p_attention_dropout
 
 p_decoder_dropout
 
 prenet
 
 attention_rnn
 
 attention_layer
 
 decoder_rnn
 
 linear_projection
 
 gate_layer
 

Constructor & Destructor Documentation

◆ __init__()

def export_tacotron2_onnx.DecoderIter.__init__ (   self,
  tacotron2 
)

Member Function Documentation

◆ decode()

def export_tacotron2_onnx.DecoderIter.decode (   self,
  decoder_input,
  in_attention_hidden,
  in_attention_cell,
  in_decoder_hidden,
  in_decoder_cell,
  in_attention_weights,
  in_attention_weights_cum,
  in_attention_context,
  memory,
  processed_memory,
  mask 
)
Here is the caller graph for this function:

◆ forward()

def export_tacotron2_onnx.DecoderIter.forward (   self,
  decoder_input,
  attention_hidden,
  attention_cell,
  decoder_hidden,
  decoder_cell,
  attention_weights,
  attention_weights_cum,
  attention_context,
  memory,
  processed_memory,
  mask 
)
Here is the call graph for this function:

Member Data Documentation

◆ tacotron2

export_tacotron2_onnx.DecoderIter.tacotron2

◆ p_attention_dropout

export_tacotron2_onnx.DecoderIter.p_attention_dropout

◆ p_decoder_dropout

export_tacotron2_onnx.DecoderIter.p_decoder_dropout

◆ prenet

export_tacotron2_onnx.DecoderIter.prenet

◆ attention_rnn

export_tacotron2_onnx.DecoderIter.attention_rnn

◆ attention_layer

export_tacotron2_onnx.DecoderIter.attention_layer

◆ decoder_rnn

export_tacotron2_onnx.DecoderIter.decoder_rnn

◆ linear_projection

export_tacotron2_onnx.DecoderIter.linear_projection

◆ gate_layer

export_tacotron2_onnx.DecoderIter.gate_layer

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