onnxruntime/js/web/test/data/ops/concat.jsonc
2021-08-12 12:30:49 -07:00

410 lines
11 KiB
Text

[
{
"name": "Concat 2D axis=0",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 0, "type": "int" }],
"cases": [
{
"name": "[4,4]",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16],
"dims": [4, 4],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16],
"dims": [4, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 1, 2, 3, 4,
5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16
],
"dims": [8, 4],
"type": "float32"
}
]
},
{
"name": "[2,4]",
"inputs": [
{
"data": [1, 2, 5, 6, 3, 4, 7, 8],
"dims": [2, 4],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 1, 2, 3, 4, 5, 6, 7, 8],
"dims": [4, 4],
"type": "float32"
}
]
},
{
"name": "[2,3]",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
}
],
"outputs": [
{
"data": [1, 2, 3, 4, 5, 6, 1, 2, 3, 4, 5, 6],
"dims": [4, 3],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 2D axis=1",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 1, "type": "int" }],
"cases": [
{
"name": "[4,4]",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16],
"dims": [4, 4],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16],
"dims": [4, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 3, 4, 1, 2, 3, 4, 5, 6, 7, 8, 5, 6, 7, 8, 9, 10, 11, 12, 9,
10, 11, 12, 13, 14, 15, 16, 13, 14, 15, 16
],
"dims": [4, 8],
"type": "float32"
}
]
},
{
"name": "[2,4]",
"inputs": [
{
"data": [1, 2, 5, 6, 3, 4, 7, 8],
"dims": [2, 4],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6, 7, 8],
"dims": [2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [1, 2, 5, 6, 1, 2, 3, 4, 3, 4, 7, 8, 5, 6, 7, 8],
"dims": [2, 8],
"type": "float32"
}
]
},
{
"name": "[2,3]",
"inputs": [
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
},
{
"data": [1, 2, 3, 4, 5, 6],
"dims": [2, 3],
"type": "float32"
}
],
"outputs": [
{
"data": [1, 2, 3, 1, 2, 3, 4, 5, 6, 4, 5, 6],
"dims": [2, 6],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 3D axis=0",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 0, "type": "int" }],
"cases": [
{
"name": "[2,2,4]",
"inputs": [
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16],
"dims": [2, 2, 4],
"type": "float32"
},
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16],
"dims": [2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 1, 2, 5, 6,
3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16
],
"dims": [4, 2, 4],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 3D axis=1",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 1, "type": "int" }],
"cases": [
{
"name": "[2,2,4]",
"inputs": [
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16],
"dims": [2, 2, 4],
"type": "float32"
},
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16],
"dims": [2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11,
12, 15, 16, 9, 10, 13, 14, 11, 12, 15, 16
],
"dims": [2, 4, 4],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 3D axis=2",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 2, "type": "int" }],
"cases": [
{
"name": "[2,2,4]",
"inputs": [
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16],
"dims": [2, 2, 4],
"type": "float32"
},
{
"data": [1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16],
"dims": [2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 1, 2, 5, 6, 3, 4, 7, 8, 3, 4, 7, 8, 9, 10, 13, 14, 9,
10, 13, 14, 11, 12, 15, 16, 11, 12, 15, 16
],
"dims": [2, 2, 8],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 4D axis=0",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 0, "type": "int" }],
"cases": [
{
"name": "[2,2,2,4]",
"inputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
},
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32, 1, 2, 5, 6, 3,
4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21, 22, 19, 20,
23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [4, 2, 2, 4],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 4D axis=1",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 1, "type": "int" }],
"cases": [
{
"name": "[2,2,2,4]",
"inputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
},
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 1, 2, 5, 6,
3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21, 22, 19, 20,
23, 24, 25, 26, 29, 30, 27, 28, 31, 32, 17, 18, 21, 22, 19, 20,
23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 4, 2, 4],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 4D axis=2",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 2, "type": "int" }],
"cases": [
{
"name": "[2,2,2,4]",
"inputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
},
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11,
12, 15, 16, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21, 22, 19, 20,
23, 24, 17, 18, 21, 22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28,
31, 32, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 4, 4],
"type": "float32"
}
]
}
]
},
{
"name": "Concat 4D axis=3",
"operator": "Concat",
"attributes": [{ "name": "axis", "data": 3, "type": "int" }],
"cases": [
{
"name": "[2,2,2,4]",
"inputs": [
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
},
{
"data": [
1, 2, 5, 6, 3, 4, 7, 8, 9, 10, 13, 14, 11, 12, 15, 16, 17, 18, 21,
22, 19, 20, 23, 24, 25, 26, 29, 30, 27, 28, 31, 32
],
"dims": [2, 2, 2, 4],
"type": "float32"
}
],
"outputs": [
{
"data": [
1, 2, 5, 6, 1, 2, 5, 6, 3, 4, 7, 8, 3, 4, 7, 8, 9, 10, 13, 14, 9,
10, 13, 14, 11, 12, 15, 16, 11, 12, 15, 16, 17, 18, 21, 22, 17,
18, 21, 22, 19, 20, 23, 24, 19, 20, 23, 24, 25, 26, 29, 30, 25,
26, 29, 30, 27, 28, 31, 32, 27, 28, 31, 32
],
"dims": [2, 2, 2, 8],
"type": "float32"
}
]
}
]
}
]