Functions | |
def | parse_args (parser) |
def | init_decoder_inputs (memory, processed_memory, memory_lengths) |
def | init_decoder_outputs (memory, memory_lengths) |
def | init_decoder_tensors (decoder_inputs, decoder_outputs) |
def | swap_inputs_outputs (decoder_inputs, decoder_outputs) |
def | infer_tacotron2_trt (encoder, decoder_iter, postnet, encoder_context, decoder_context, postnet_context, sequences, sequence_lengths, measurements, fp16) |
def | infer_waveglow_trt (waveglow, waveglow_context, mel, measurements, fp16) |
def | main () |
def inference_trt.parse_args | ( | parser | ) |
Parse commandline arguments.
def inference_trt.init_decoder_inputs | ( | memory, | |
processed_memory, | |||
memory_lengths | |||
) |
def inference_trt.init_decoder_outputs | ( | memory, | |
memory_lengths | |||
) |
def inference_trt.init_decoder_tensors | ( | decoder_inputs, | |
decoder_outputs | |||
) |
def inference_trt.swap_inputs_outputs | ( | decoder_inputs, | |
decoder_outputs | |||
) |
def inference_trt.infer_tacotron2_trt | ( | encoder, | |
decoder_iter, | |||
postnet, | |||
encoder_context, | |||
decoder_context, | |||
postnet_context, | |||
sequences, | |||
sequence_lengths, | |||
measurements, | |||
fp16 | |||
) |
def inference_trt.infer_waveglow_trt | ( | waveglow, | |
waveglow_context, | |||
mel, | |||
measurements, | |||
fp16 | |||
) |
def inference_trt.main | ( | ) |