tf.keras.quantizers.compute_float8_scale