trt_utils Namespace Reference

Functions
def	is_dimension_dynamic (dim)
def	is_shape_dynamic (shape)
def	run_trt_engine (context, engine, tensors)
def	load_engine (engine_filepath, trt_logger)
def	engine_info (engine_filepath)
def	build_engine (model_file, shapes, max_ws=512 *1024 *1024, fp16=False)

Function Documentation

is_dimension_dynamic()

def trt_utils.is_dimension_dynamic

(

dim

)

Here is the caller graph for this function:

is_shape_dynamic()

def trt_utils.is_shape_dynamic

(

shape

)

Here is the call graph for this function:

Here is the caller graph for this function:

run_trt_engine()

def trt_utils.run_trt_engine	(		context,
			engine,
			tensors
	)

Here is the call graph for this function:

load_engine()

def trt_utils.load_engine	(		engine_filepath,
			trt_logger
	)

Here is the caller graph for this function:

engine_info()

def trt_utils.engine_info

(

engine_filepath

)

Here is the call graph for this function:

build_engine()

def trt_utils.build_engine	(		model_file,
			shapes,
			max_ws = 512*1024*1024,
			fp16 = False
	)

Here is the call graph for this function: