dnn_backend_native_layer_mathbinary: add floormod support

Mingyu Yin committed 6 years ago

Commit 3477feb1 parent 37ef1ac5 files · +15 −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

@@ -73,7 +73,7 @@ class TFConverter: · 72 unmodified lines
73	73	self.conv2d_scopename_inputname_dict = {}
74	74	self.op2code = {'Conv2D':1, 'DepthToSpace':2, 'MirrorPad':3, 'Maximum':4,
75	75	'MathBinary':5, 'MathUnary':6, 'AvgPool':7}
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, 'FloorMod':5}
77	77	self.mathun2code = {'Abs':0, 'Sin':1, 'Cos':2, 'Tan':3, 'Asin':4,
78	78	'Acos':5, 'Atan':6, 'Sinh':7, 'Cosh':8, 'Tanh':9, 'Asinh':10,
79	79	'Acosh':11, 'Atanh':12, 'Ceil':13, 'Floor':14, 'Round':15}

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