mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
a2e9ba72d5
### Description Added FusedConv and FusedConvTranspose ### Motivation and Context <!-- - Why is this change required? What problem does it solve? - If it fixes an open issue, please link to the issue here. --> Improve performance
112 lines
2.6 KiB
Text
112 lines
2.6 KiB
Text
[
|
|
{
|
|
"name": "conv without bias addition A",
|
|
"operator": "FusedConv",
|
|
"attributes": [
|
|
{ "name": "activation", "data": "Relu", "type": "string" },
|
|
{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }
|
|
],
|
|
"opset": { "domain": "com.microsoft", "version": 1 },
|
|
"cases": [
|
|
{
|
|
"name": "T[0]",
|
|
"inputs": [
|
|
{
|
|
"data": [10, 20, 30, 40, 50, 60, 70, 80, 90],
|
|
"dims": [1, 1, 3, 3],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 2, 3, 4],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [370, 470, 670, 770],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "T[1]",
|
|
"inputs": [
|
|
{
|
|
"data": [10, 20, -30, -40, -50, -60, 70, 80, 90],
|
|
"dims": [1, 1, 3, 3],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 2, 3, 4],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [0, 0, 390, 430],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "NHWC conv without bias addition A",
|
|
"operator": "Conv",
|
|
"attributes": [
|
|
{ "name": "activation", "data": "Relu", "type": "string" },
|
|
{ "name": "kernel_shape", "data": [2, 2], "type": "ints" }
|
|
],
|
|
"opset": { "domain": "com.ms.internal.nhwc", "version": 11 },
|
|
"cases": [
|
|
{
|
|
"name": "T[2]",
|
|
"inputs": [
|
|
{
|
|
"data": [10, 20, 30, 40, 50, 60, 70, 80, 90],
|
|
"dims": [1, 3, 3, 1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 2, 3, 4],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [370, 470, 670, 770],
|
|
"dims": [1, 2, 2, 1],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "T[3]",
|
|
"inputs": [
|
|
{
|
|
"data": [10, 20, -30, -40, -50, -60, 70, 80, 90],
|
|
"dims": [1, 3, 3, 1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 2, 3, 4],
|
|
"dims": [1, 1, 2, 2],
|
|
"type": "float32"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [0, 0, 390, 430],
|
|
"dims": [1, 2, 2, 1],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|