def testMaxPool(self):
y = extensions.max_pool(np.ones([5, 320, 480, 3]), [3, 5], [2, 3],
"VALID")
self.assertAllEqual(
y,
tf.nn.pool(
input=tf.ones([5, 320, 480, 3]),
window_shape=[3, 5],
pooling_type="MAX",
padding="VALID",
strides=[2, 3],
))
def testMaxPool(self):
y = extensions.max_pool(np.ones([5, 320, 480, 3]), [3, 5], [2, 3], "VALID")
self.assertAllEqual(
y, tf.nn.pool(tf.ones([5, 320, 480, 3]), [3, 5], "MAX", "VALID",
strides=[2, 3],))
