Skip to content

Commit e5e151d

Browse files
authored
[tools/onnx-subgraph] add config.json for onnx model splitting (#14852)
config.json for onnx model splitting, it will be used in future PRs ONE-DCO-1.0-Signed-off-by: Youxin Chen <yx113.chen@samsung.com>
1 parent 8035ab0 commit e5e151d

File tree

3 files changed

+39
-0
lines changed

3 files changed

+39
-0
lines changed
Lines changed: 13 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,13 @@
1+
{
2+
"NPU_supported_ops": ["Conv", "Reshape", "Transpose", "Add", "ReduceMean", "Sub",
3+
"Div", "Mul", "Sigmoid","MatMul"],
4+
"CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose",
5+
"Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape",
6+
"Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand",
7+
"Slice", "Unsqueeze"],
8+
"performance_data": [],
9+
"hardware_limits": {
10+
"max_subgraph_size": 60024.0,
11+
"max_subgraphs": 5
12+
}
13+
}
Lines changed: 12 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,12 @@
1+
{
2+
"NPU_supported_ops": [],
3+
"CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose",
4+
"Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape",
5+
"Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand",
6+
"Slice", "Unsqueeze"],
7+
"performance_data": [],
8+
"hardware_limits": {
9+
"max_subgraph_size": 60024.0,
10+
"max_subgraphs": 5
11+
}
12+
}
Lines changed: 14 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,14 @@
1+
{
2+
"NPU_supported_ops": ["Conv", "Reshape", "Transpose", "Add", "ReduceMean", "Sub",
3+
"Div", "Mul", "Sigmoid","MatMul"],
4+
"CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose",
5+
"Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape",
6+
"Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand",
7+
"Slice", "Unsqueeze"],
8+
"performance_data": [{"name":"Conv","CPU_time": 0.1, "NPU_time": 0.05},
9+
{"name":"Mul", "CPU_time": 0.15, "NPU_time": 0.07}],
10+
"hardware_limits": {
11+
"max_subgraph_size": 60024.0,
12+
"max_subgraphs": 5
13+
}
14+
}

0 commit comments

Comments
 (0)