Showing the changes to tools/python/convert_from_tensorflow.py (1 of 5 files). Show all
Unified · Split
| @@ -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} |