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 Namespace Reference

Classes

class  DecoderIter
 
class  Encoder
 
class  Postnet
 

Functions

def parse_args (parser)
 
def encoder_infer (self, x, input_lengths)
 
def lstmcell2lstm_params (lstm_mod, lstmcell_mod)
 
def prenet_infer (self, x)
 
def test_inference (encoder, decoder_iter, postnet)
 
def main ()
 

Function Documentation

◆ parse_args()

def export_tacotron2_onnx.parse_args (   parser)
Parse commandline arguments.
Here is the caller graph for this function:

◆ encoder_infer()

def export_tacotron2_onnx.encoder_infer (   self,
  x,
  input_lengths 
)

◆ lstmcell2lstm_params()

def export_tacotron2_onnx.lstmcell2lstm_params (   lstm_mod,
  lstmcell_mod 
)

◆ prenet_infer()

def export_tacotron2_onnx.prenet_infer (   self,
  x 
)

◆ test_inference()

def export_tacotron2_onnx.test_inference (   encoder,
  decoder_iter,
  postnet 
)
Here is the caller graph for this function:

◆ main()

def export_tacotron2_onnx.main ( )
Here is the call graph for this function: