import tensorflow as tf

            X = tf.nn.relu(X)
        X = tf.nn.conv2d(X, W, (1, 1, 1, 1), padding='SAME')
        X = self._add_batch_norm(X, out_ch, is_train=is_train)
        X = tf.reshape(X, (-1, w, h, out_ch)) # Sanity shape check
        return X

