tf.keras.utils.convert_all_kernels_in_model