fix quantize_annotate_layer() got an unexpected keyword argument 'input_shape'

Author: moneypi

tensorflow_model_optimization/python/examples/quantization/keras/mnist_cnn.py

@@ -59,8 +59,7 @@
 
 model = tf.keras.Sequential([
     quantize.quantize_annotate_layer(
-        l.Conv2D(32, 5, padding='same', activation='relu'),
-        input_shape=input_shape),
+        l.Conv2D(32, 5, padding='same', activation='relu', input_shape=input_shape)),
     l.MaxPooling2D((2, 2), (2, 2), padding='same'),
     quantize.quantize_annotate_layer(
         l.Conv2D(64, 5, padding='same', activation='relu')),