tf.keras.quantizers.compute_float8_amax_history