dnn_backend_native_layer_mathunary: add tanh support

Ting Fu committed 6 years ago

Commit ea71e731 parent c5de77e4 files · +7 −2Browse files

Showing the changes to tools/python (2 of 4 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

@@ -72,7 +72,7 @@ class TFConverter: · 71 unmodified lines
72	72	self.conv2d_scopename_inputname_dict = {}
73	73	self.op2code = {'Conv2D':1, 'DepthToSpace':2, 'MirrorPad':3, 'Maximum':4, 'MathBinary':5, 'MathUnary':6}
74	74	self.mathbin2code = {'Sub':0, 'Add':1, 'Mul':2, 'RealDiv':3, 'Minimum':4}
75		− self.mathun2code = {'Abs':0, 'Sin':1, 'Cos':2, 'Tan':3, 'Asin':4, 'Acos':5, 'Atan':6, 'Sinh':7, 'Cosh':8}
	75	+ self.mathun2code = {'Abs':0, 'Sin':1, 'Cos':2, 'Tan':3, 'Asin':4, 'Acos':5, 'Atan':6, 'Sinh':7, 'Cosh':8, 'Tanh':9}
76	76	self.mirrorpad_mode = {'CONSTANT':0, 'REFLECT':1, 'SYMMETRIC':2}
77	77	self.name_operand_dict = {}
78	78

tools/python/convert_header.py+1−1

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