def testSeparableConv(self): x = np.random.rand(5, 7, 1, 11) y = common_layers.separable_conv( tf.constant(x, dtype=tf.float32), 13, (3, 1)) self.evaluate(tf.global_variables_initializer()) res = self.evaluate(y) self.assertEqual(res.shape, (5, 5, 1, 13))
def testSeparableConv(self): x = np.random.rand(5, 7, 1, 11) y = common_layers.separable_conv(tf.constant(x, dtype=tf.float32), 13, (3, 1)) self.evaluate(tf.global_variables_initializer()) res = self.evaluate(y) self.assertEqual(res.shape, (5, 5, 1, 13))