mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-27 22:45:57 +00:00
841f7ed3e0
### Description <!-- Describe your changes. --> Added Uniforms to Expand operator kernel ### 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
118 lines
2.2 KiB
Text
118 lines
2.2 KiB
Text
[
|
|
{
|
|
"name": "Expand 1 - float32",
|
|
"operator": "Expand",
|
|
"attributes": [],
|
|
"cases": [
|
|
{
|
|
"name": "Expand 1 - float32",
|
|
"inputs": [
|
|
{
|
|
"data": [1],
|
|
"dims": [1, 1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 4],
|
|
"dims": [2],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1, 1, 1, 1],
|
|
"dims": [1, 4],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Expand 2 - float32",
|
|
"operator": "Expand",
|
|
"attributes": [],
|
|
"cases": [
|
|
{
|
|
"name": "Expand 2 - float32",
|
|
"inputs": [
|
|
{
|
|
"data": [1],
|
|
"dims": [1, 1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 6],
|
|
"dims": [2],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1, 1, 1, 1, 1, 1],
|
|
"dims": [1, 6],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Expand 1D - float32",
|
|
"operator": "Expand",
|
|
"attributes": [],
|
|
"cases": [
|
|
{
|
|
"name": "Expand 1D - float32",
|
|
"inputs": [
|
|
{
|
|
"data": [1],
|
|
"dims": [1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [4],
|
|
"dims": [1],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1, 1, 1, 1],
|
|
"dims": [4],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Expand 5D - float32",
|
|
"operator": "Expand",
|
|
"attributes": [],
|
|
"cases": [
|
|
{
|
|
"name": "Expand 5 - float32",
|
|
"inputs": [
|
|
{
|
|
"data": [1],
|
|
"dims": [1, 1, 1, 1, 1],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [1, 1, 1, 1, 6],
|
|
"dims": [5],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1, 1, 1, 1, 1, 1],
|
|
"dims": [1, 1, 1, 1, 6],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|