当使用Tensorflow优化器而不是Keras优化器时，Keras模型会显着降低速度

Question

对于我的keras模型，我需要使用tensorflow优化器来支持TPU。但是，当我使用optimizer=tf.train.RMSPropOptimizer(learning_rate)而不是optimizer=tf.keras.optimizers.RMSprop(lr=learning_rate)时，模型速度降低了7倍。

模型tf.keras，并使用一个嵌入层，一些Lambda层和一个合并层。

我该怎么办？有什么想法吗？