多组算子描述文件配置
描述文件支持定义多组算子JSON文件配置,一组配置包括算子类型、算子输入和输出信息、视算子情况决定是否包括属性信息。
如果JSON文件配置了多组算子,则模型转换完成后,会生成多组算子对应的om离线模型文件。如下配置文件只是样例,请根据实际情况进行修改。
[
{
"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
},
... ...
]
}
]
父主题: 配置文件样例