convert_weights Namespace Reference

Functions
def	float_to_hex (f)

Variables
	parser = argparse.ArgumentParser(description='TensorFlow to TensorRT Weight Dumper')
	required
	True
	help
	opt = parser.parse_args()
	inputbase = opt.model
	outputbase = opt.output
dictionary	remap

Function Documentation

float_to_hex()

def convert_weights.float_to_hex

(

f

)

Variable Documentation

parser

convert_weights.parser = argparse.ArgumentParser(description='TensorFlow to TensorRT Weight Dumper')

required

convert_weights.required

True

convert_weights.True

help

convert_weights.help

opt

convert_weights.opt = parser.parse_args()

inputbase

convert_weights.inputbase = opt.model

outputbase

convert_weights.outputbase = opt.output

remap

dictionary convert_weights.remap

Initial value:

=  {
    'Variable': 'hiddenWeights0',
    'Variable_1': 'hiddenWeights1',
    'Variable_2': 'outputWeights',
    'Variable_3': 'hiddenBias0',
    'Variable_4': 'hiddenBias1',
    'Variable_5': 'outputBias'
}