Source code
Revision control
Copy as Markdown
Other Tools
{
"tests": [
{
"name": "split float32 1D constant tensor number splits default options",
"inputs": {
"input": {
"shape": [24],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32",
"constant": true
}
},
"splits": 3,
"expected": [
{
"name": "output1",
"shape": [8],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094
],
"type": "float32"
},
{
"name": "output2",
"shape": [8],
"data": [
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438,
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422
],
"type": "float32"
},
{
"name": "output3",
"shape": [8],
"data": [
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 1D tensor number splits default options",
"inputs": {
"input": {
"shape": [24],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": 3,
"expected": [
{
"name": "output1",
"shape": [8],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094
],
"type": "float32"
},
{
"name": "output2",
"shape": [8],
"data": [
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438,
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422
],
"type": "float32"
},
{
"name": "output3",
"shape": [8],
"data": [
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 2D tensor number splits default options",
"inputs": {
"input": {
"shape": [8, 3],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": 2,
"expected": [
{
"name": "output1",
"shape": [4, 3],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094,
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438
],
"type": "float32"
},
{
"name": "output2",
"shape": [4, 3],
"data": [
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422,
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 3D tensor number splits default options",
"inputs": {
"input": {
"shape": [4, 3, 2],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": 2,
"expected": [
{
"name": "output1",
"shape": [2, 3, 2],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094,
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438
],
"type": "float32"
},
{
"name": "output2",
"shape": [2, 3, 2],
"data": [
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422,
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 4D tensor number splits default options",
"inputs": {
"input": {
"shape": [12, 1, 1, 2],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": 4,
"expected": [
{
"name": "output1",
"shape": [3, 1, 1, 2],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258
],
"type": "float32"
},
{
"name": "output2",
"shape": [3, 1, 1, 2],
"data": [
-68.11224365234375,
75.99607849121094,
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438
],
"type": "float32"
},
{
"name": "output3",
"shape": [3, 1, 1, 2],
"data": [
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422,
50.66357421875,
21.364582061767578
],
"type": "float32"
},
{
"name": "output4",
"shape": [3, 1, 1, 2],
"data": [
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 5D tensor number splits default options",
"inputs": {
"input": {
"shape": [6, 1, 1, 2, 2],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": 2,
"expected": [
{
"name": "output1",
"shape": [3, 1, 1, 2, 2],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094,
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438
],
"type": "float32"
},
{
"name": "output2",
"shape": [3, 1, 1, 2, 2],
"data": [
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422,
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 4D tensor array splits default options",
"inputs": {
"input": {
"shape": [12, 1, 1, 2],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": [3, 3, 3, 3],
"expected": [
{
"name": "output1",
"shape": [3, 1, 1, 2],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258
],
"type": "float32"
},
{
"name": "output2",
"shape": [3, 1, 1, 2],
"data": [
-68.11224365234375,
75.99607849121094,
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438
],
"type": "float32"
},
{
"name": "output3",
"shape": [3, 1, 1, 2],
"data": [
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422,
50.66357421875,
21.364582061767578
],
"type": "float32"
},
{
"name": "output4",
"shape": [3, 1, 1, 2],
"data": [
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 4D tensor number splits options.axis",
"inputs": {
"input": {
"shape": [12, 1, 1, 2],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": 3,
"options": {
"axis": 0
},
"expected": [
{
"name": "output1",
"shape": [4, 1, 1, 2],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094
],
"type": "float32"
},
{
"name": "output2",
"shape": [4, 1, 1, 2],
"data": [
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438,
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422
],
"type": "float32"
},
{
"name": "output3",
"shape": [4, 1, 1, 2],
"data": [
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
},
{
"name": "split float32 5D tensor array splits options.axis",
"inputs": {
"input": {
"shape": [1, 1, 6, 2, 2],
"data": [
-64.52057192772567,
-84.60513454654502,
-67.9928282139549,
-23.446074622313745,
-85.64383132426956,
46.877522730892906,
-68.11224159891194,
75.99607609082298,
-61.05668616435924,
-90.92643605408041,
53.916621140775476,
84.16269171442175,
-95.57494716023527,
-52.40757884637803,
-29.007187148127983,
71.65495730084652,
50.663572707062656,
21.364581604631596,
-27.127241457105228,
65.1489516233315,
-30.406812651567833,
-6.818390152438795,
46.67362179020569,
-21.12453802903387
],
"type": "float32"
}
},
"splits": [3, 3],
"options": {
"axis": 2
},
"expected": [
{
"name": "output1",
"shape": [1, 1, 3, 2, 2],
"data": [
-64.52056884765625,
-84.60513305664062,
-67.99282836914062,
-23.446075439453125,
-85.64382934570312,
46.87752151489258,
-68.11224365234375,
75.99607849121094,
-61.05668640136719,
-90.92643737792969,
53.916622161865234,
84.16268920898438
],
"type": "float32"
},
{
"name": "output2",
"shape": [1, 1, 3, 2, 2],
"data": [
-95.57494354248047,
-52.40757751464844,
-29.007186889648438,
71.65496063232422,
50.66357421875,
21.364582061767578,
-27.127241134643555,
65.1489486694336,
-30.40681266784668,
-6.818390369415283,
46.673622131347656,
-21.12453842163086
],
"type": "float32"
}
]
}
]
}