{
  "tests": [
    {
      "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 4D tensor number splits negative options.axis",
      "inputs": {
        "input": {
          "shape": [1, 1, 12, 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,
      "options": {
        "axis": -2
      },
      "expected": [
        {
          "name": "output1",
          "shape": [1, 1, 6, 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, 6, 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 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"
        }
      ]
    },
    {
      "name": "split float32 5D tensor array splits negative options.axis",
      "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, 4],
      "options": {
        "axis": -5
      },
      "expected": [
        {
          "name": "output1",
          "shape": [2, 1, 1, 2, 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, 2],
          "data": [
            -61.05668640136719,
            -90.92643737792969,
            53.916622161865234,
            84.16268920898438,
            -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"
        }
      ]
    }
  ]
}