替换现有keras（tensorflow）模型中的乘法算子

我目前正在使用某个模型的现有Keras实现，我想研究不同乘法实现对其计算速度和精度的影响

有没有一种简单的方法可以用自定义的Keras（TensorFlow）乘法替换密集层和Conv（以及其他预先存在的）层中使用的Keras（TensorFlow）乘法

我们的想法还在于了解使用常规乘法进行训练+使用自定义乘法进行测试与同时使用自定义乘法进行两者之间的区别

因此，我正在寻找一种类似于：

import tensorflow as tf
tf.__mul__ = custom_mult

并将用我自己的实现替换Keras默认层中的所有乘法运算