import tensorflow as tf

        with self.graph.as_default():
            with tf.device('/cpu:0'):
                embedding = tf.get_variable(
                    'embedding', [vocab_size, hidden_size], dtype=tf.float32)
