This seems to be a limitation with the Vulkan API, unfortunately there isn’t any workaround for it. Although there may technically be enough memory on the device to store the tensor data, whatever propriety technique used to represent textures on the device probably can’t handle textures of that size. A solution would involve using multiple textures to represent one tensor if the dimensions exceed the limit, which we don’t plan to implement at the moment.
Just curious, what GPU/device are you using?