lavfi/dnn/dnn_backend_native_layer_mathunary: add exp support

Wenlong Ding committed 5 years ago

Commit b4605951 parent dfad5794 files · +8 −2Browse files

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

Unified · Split

2 files

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

tools/python/convert_from_tensorflow.py+2−1

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

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 = 22	26	minor = 23