tf.keras.layers.MaxPool2D