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

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 ()
 

Function Documentation

◆ parse_args()

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

◆ init_decoder_inputs()

def inference_trt.init_decoder_inputs (   memory,
  processed_memory,
  memory_lengths 
)
Here is the caller graph for this function:

◆ init_decoder_outputs()

def inference_trt.init_decoder_outputs (   memory,
  memory_lengths 
)
Here is the caller graph for this function:

◆ init_decoder_tensors()

def inference_trt.init_decoder_tensors (   decoder_inputs,
  decoder_outputs 
)
Here is the caller graph for this function:

◆ swap_inputs_outputs()

def inference_trt.swap_inputs_outputs (   decoder_inputs,
  decoder_outputs 
)
Here is the caller graph for this function:

◆ infer_tacotron2_trt()

def inference_trt.infer_tacotron2_trt (   encoder,
  decoder_iter,
  postnet,
  encoder_context,
  decoder_context,
  postnet_context,
  sequences,
  sequence_lengths,
  measurements,
  fp16 
)
Here is the call graph for this function:
Here is the caller graph for this function:

◆ infer_waveglow_trt()

def inference_trt.infer_waveglow_trt (   waveglow,
  waveglow_context,
  mel,
  measurements,
  fp16 
)
Here is the caller graph for this function:

◆ main()

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