Added support for the "Split" op and added a VJP for the casting initializer. (#24120)
* Added support for the split op.
* Minor edit.
* Added VJP for the casting initializer.
* Update stdlib/public/TensorFlow/Tensor.swift
Co-Authored-By: eaplatanios <e.a.platanios@gmail.com>
* Update stdlib/public/TensorFlow/Tensor.swift
Co-Authored-By: eaplatanios <e.a.platanios@gmail.com>