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

Functions

def float_to_hex (f)
 
def getTRTType (tensor)
 

Variables

 parser = argparse.ArgumentParser(description='TensorFlow Weight Dumper')
 
 required
 
 True
 
 help
 
 False
 
 default
 
 type
 
 bool
 
 opt = parser.parse_args()
 
 inputbase = opt.model
 
 outputbase = opt.output
 
string proposed_file = ".".join(inputbase.split(".")[0:-1])
 
string v2_file_error_template
 

Function Documentation

◆ float_to_hex()

def dumpTFWts.float_to_hex (   f)
Here is the caller graph for this function:

◆ getTRTType()

def dumpTFWts.getTRTType (   tensor)
Here is the call graph for this function:

Variable Documentation

◆ parser

dumpTFWts.parser = argparse.ArgumentParser(description='TensorFlow Weight Dumper')

◆ required

dumpTFWts.required

◆ True

dumpTFWts.True

◆ help

dumpTFWts.help

◆ False

dumpTFWts.False

◆ default

dumpTFWts.default

◆ type

dumpTFWts.type

◆ bool

dumpTFWts.bool

◆ opt

dumpTFWts.opt = parser.parse_args()

◆ inputbase

dumpTFWts.inputbase = opt.model

◆ outputbase

dumpTFWts.outputbase = opt.output

◆ proposed_file

string dumpTFWts.proposed_file = ".".join(inputbase.split(".")[0:-1])

◆ v2_file_error_template

string dumpTFWts.v2_file_error_template
Initial value:
1 = """
2  It's likely that this is a V2 checkpoint and you need to provide the filename
3  *prefix*. Try removing the '.' and extension. Try:
4  inspect checkpoint --file_name = {}"""