import tensorflow as tf

            pre_pool_out = sess.run(pre_pool, feed_dict=feed_dict)
            self.assertListEqual(list(pre_pool_out.shape), [batch_size, 7, 7, 1024])

    def testGlobalPoolUnknownImageShape(self):
        tf.reset_default_graph()
        batch_size = 1
        height, width = 250, 300
        num_classes = 1000
