diff --git a/tools/onnx_subgraph/examples/config-cpunpu.json b/tools/onnx_subgraph/examples/config-cpunpu.json new file mode 100644 index 00000000000..700ea499bfa --- /dev/null +++ b/tools/onnx_subgraph/examples/config-cpunpu.json @@ -0,0 +1,13 @@ +{ + "NPU_supported_ops": ["Conv", "Reshape", "Transpose", "Add", "ReduceMean", "Sub", + "Div", "Mul", "Sigmoid","MatMul"], + "CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose", + "Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape", + "Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand", + "Slice", "Unsqueeze"], + "performance_data": [], + "hardware_limits": { + "max_subgraph_size": 60024.0, + "max_subgraphs": 5 + } +} diff --git a/tools/onnx_subgraph/examples/config-cpuonly.json b/tools/onnx_subgraph/examples/config-cpuonly.json new file mode 100644 index 00000000000..7f408d6a4fd --- /dev/null +++ b/tools/onnx_subgraph/examples/config-cpuonly.json @@ -0,0 +1,12 @@ +{ + "NPU_supported_ops": [], + "CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose", + "Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape", + "Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand", + "Slice", "Unsqueeze"], + "performance_data": [], + "hardware_limits": { + "max_subgraph_size": 60024.0, + "max_subgraphs": 5 + } +} diff --git a/tools/onnx_subgraph/examples/config-full.json b/tools/onnx_subgraph/examples/config-full.json new file mode 100644 index 00000000000..b5d15f800a5 --- /dev/null +++ b/tools/onnx_subgraph/examples/config-full.json @@ -0,0 +1,14 @@ +{ + "NPU_supported_ops": ["Conv", "Reshape", "Transpose", "Add", "ReduceMean", "Sub", + "Div", "Mul", "Sigmoid","MatMul"], + "CPU_supported_ops": ["Sub", "Pow", "ReduceMean", "Add", "Sqrt", "Div","Transpose", + "Gather", "MatMul", "Mul", "Softmax", "Erf", "Gemm", "Conv", "Reshape", + "Sin", "Where", "ConstantOfShape", "Cast", "Sigmoid", "Cos", "Expand", + "Slice", "Unsqueeze"], + "performance_data": [{"name":"Conv","CPU_time": 0.1, "NPU_time": 0.05}, + {"name":"Mul", "CPU_time": 0.15, "NPU_time": 0.07}], + "hardware_limits": { + "max_subgraph_size": 60024.0, + "max_subgraphs": 5 + } +}