def testConvStride2MultiStep(self): x1 = np.random.rand(5, 32, 16, 11) a = common_layers.conv_stride2_multistep( tf.constant(x1, dtype=tf.float32), 4, 16) self.evaluate(tf.global_variables_initializer()) actual = self.evaluate(a[0]) self.assertEqual(actual.shape, (5, 2, 1, 16))