def METHOD_NAME():
    valid_input = tf.keras.layers.Input((None, None, 1))
    updates, indices = tf.nn.max_pool_with_argmax(
        valid_input, ksize=[2, 2], strides=[2, 2], padding="SAME"
    )
    with pytest.raises(ValueError):
        MaxUnpooling2DV2(valid_input.shape)(updates, indices)