def test_add_edge_bias(self): with tf.Graph().as_default(): x = tf.random_uniform(shape=(16, 32, 32, 3)) x_pad = glow_ops.add_edge_bias(x, [3, 3]) with tf.Session() as session: x_pad_np = session.run(x_pad) # Test expected output shape. self.assertEqual(x_pad_np.shape, (16, 34, 34, 4))
def test_add_edge_bias(self): with tf.Graph().as_default(): x = tf.random_uniform(shape=(16, 32, 32, 3)) x_pad = glow_ops.add_edge_bias(x, [3, 3]) with tf.Session() as session: x_pad_np = session.run(x_pad) # Test expected output shape. self.assertEqual(x_pad_np.shape, (16, 34, 34, 4))