tf.keras.quantizers.abs_max_quantize