Tensorflow分散层

Question

我有一个30层的模型。

进行预测时，假设我要在机器A的前15层上运行预测，将第15层的输出发送到机器B，然后在机器B上运行接下来的15层。

我该怎么做？ 谢谢。