mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-16 21:00:14 +00:00
447a3a7c70
### Description Also update the op test suite. ### Motivation and Context Previously the *total* size in case `Expand - last dim is not divisible by 4` was a multiple of 4, even though the *last dimension* was not, so the bug has never been caught.
210 lines
4.2 KiB
Text
210 lines
4.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"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Expand 5 - shape < input.size()",
|
|
"inputs": [
|
|
{
|
|
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
|
|
"dims": [1, 1, 1, 2, 6],
|
|
"type": "float32"
|
|
},
|
|
{
|
|
"data": [2, 1, 6],
|
|
"dims": [3],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12],
|
|
"dims": [1, 1, 2, 2, 6],
|
|
"type": "float32"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Expand - bool",
|
|
"operator": "Expand",
|
|
"attributes": [],
|
|
"cases": [
|
|
{
|
|
"name": "Expand - last dim is divisible by 4",
|
|
"inputs": [
|
|
{
|
|
"data": [true, false, false, true],
|
|
"dims": [4],
|
|
"type": "bool"
|
|
},
|
|
{
|
|
"data": [2, 4],
|
|
"dims": [2],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [true, false, false, true, true, false, false, true],
|
|
"dims": [2, 4],
|
|
"type": "bool"
|
|
}
|
|
]
|
|
},
|
|
{
|
|
"name": "Expand - last dim is not divisible by 4",
|
|
"inputs": [
|
|
{
|
|
"data": [true, false, false, true, true, true],
|
|
"dims": [1, 6],
|
|
"type": "bool"
|
|
},
|
|
{
|
|
"data": [3, 1],
|
|
"dims": [2],
|
|
"type": "int64"
|
|
}
|
|
],
|
|
"outputs": [
|
|
{
|
|
"data": [
|
|
true,
|
|
false,
|
|
false,
|
|
true,
|
|
true,
|
|
true,
|
|
true,
|
|
false,
|
|
false,
|
|
true,
|
|
true,
|
|
true,
|
|
true,
|
|
false,
|
|
false,
|
|
true,
|
|
true,
|
|
true
|
|
],
|
|
"dims": [3, 6],
|
|
"type": "bool"
|
|
}
|
|
]
|
|
}
|
|
]
|
|
}
|
|
]
|