dnn_backend_native_layer_mathunary: add round support

Mingyu Yin committed 6 years ago

Commit 4ed6bca1 parent ed39d285 files · +11 −2Browse files

Showing the changes to tools/python (2 of 5 files). Show all

Unified · Split

2 files

tools/python/convert_from_tensorflow.py +1 −1
tools/python/convert_header.py +1 −1

tools/python/convert_from_tensorflow.py+1−1

@@ -76,7 +76,7 @@ class TFConverter: · 75 unmodified lines
76	self.mathbin2code = {'Sub':0, 'Add':1, 'Mul':2, 'RealDiv':3, 'Minimum':4}	76	self.mathbin2code = {'Sub':0, 'Add':1, 'Mul':2, 'RealDiv':3, 'Minimum':4}
77	self.mathun2code = {'Abs':0, 'Sin':1, 'Cos':2, 'Tan':3, 'Asin':4,	77	self.mathun2code = {'Abs':0, 'Sin':1, 'Cos':2, 'Tan':3, 'Asin':4,
78	'Acos':5, 'Atan':6, 'Sinh':7, 'Cosh':8, 'Tanh':9, 'Asinh':10,	78	'Acos':5, 'Atan':6, 'Sinh':7, 'Cosh':8, 'Tanh':9, 'Asinh':10,
79	'Acosh':11, 'Atanh':12, 'Ceil':13, 'Floor':14}	79	'Acosh':11, 'Atanh':12, 'Ceil':13, 'Floor':14, 'Round':15}
80	self.mirrorpad_mode = {'CONSTANT':0, 'REFLECT':1, 'SYMMETRIC':2}	80	self.mirrorpad_mode = {'CONSTANT':0, 'REFLECT':1, 'SYMMETRIC':2}
81	self.name_operand_dict = {}	81	self.name_operand_dict = {}
82		82

tools/python/convert_header.py+1−1

@@ -23,4 +23,4 @@ str = 'FFMPEGDNNNATIVE' · 22 unmodified lines
23	major = 1	23	major = 1
24		24
25	# increase minor when we don't have to re-convert the model file	25	# increase minor when we don't have to re-convert the model file
26	minor = 20	26	minor = 21