如何将tf.keras预处理层与分发策略结合

Question

Keras preprocessing layers旨在将预处理和扩充集成到模型本身中。但是，似乎这些层不能与tf.distribute.Strategy一起用于分布式培训。有没有可以在分配策略中同时使用它们的方法？