Если вы хотите развернуть модель тензорного потока в Spark, вам следует взглянуть на Deeplearning4J . Он поставляется с некоторыми импортерами, где вы можете прочитать модели keras и TensorFlow .
Помните, что поддерживается не каждый слой.
Помимо искрового глубокого обучения, есть tenorframe , я никогда не использовал его, поэтому я не знаю, насколько он хорош.
В общем, я бы предложил использовать тензорный поток напрямую через Распределенный тензорный поток и не использовать все эти оболочки.