多组算子描述文件配置
描述文件支持定义多组算子Json文件配置,一组配置包括算子类型、算子输入和输出信息、视算子情况决定是否包括属性信息。
如果Json文件配置了多组算子,则模型转换完成后,会生成多组算子对应的.om离线模型文件。如下配置文件只是样例,请根据实际情况进行修改。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
[ { "op": "MatMul", "name": "matMul01", "input_desc": [ { "format": "ND", "shape": [ 16, 16 ], "type": "float16" }, ... ], "output_desc": [ { "format": "ND", "shape": [ 16, 16 ], "type": "float16" } ], "attr": [ { "name": "alpha", "type": "float", "value": 1.0 }, ... ] }, { "op": "MatMul", "name": "matMul02", "input_desc": [ { "format": "ND", "shape": [ 256, 256 ], "type": "float16" }, ... ], "output_desc": [ { "format": "ND", "shape": [ 256, 256 ], "type": "float16" } ], "attr": [ { "name": "alpha", "type": "float", "value": 1.0 }, ... ] } ] |
父主题: 配置文件样例