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/convert_from_tensorflow.py (1 of 5 files). Show all

Unified · Split

tools/python/convert_from_tensorflow.py+1−1

@@ -73,7 +73,7 @@ class TFConverter: · 72 unmodified lines
73	self.conv2d_scopename_inputname_dict = {}	73	self.conv2d_scopename_inputname_dict = {}
74	self.op2code = {'Conv2D':1, 'DepthToSpace':2, 'MirrorPad':3, 'Maximum':4,	74	self.op2code = {'Conv2D':1, 'DepthToSpace':2, 'MirrorPad':3, 'Maximum':4,
75	'MathBinary':5, 'MathUnary':6, 'AvgPool':7}	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	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, 'Round':15}	79	'Acosh':11, 'Atanh':12, 'Ceil':13, 'Floor':14, 'Round':15}