def simple_entity_encode_op(self, params, ent):
    """Wrapper to encode a single entity into its EmQL representation."""
    set_ids = tf.expand_dims(ent, axis=1)
    set_mask = tf.ones_like(set_ids, dtype=tf.float32)
    return module.encode_op(params, set_ids, set_mask,
                            self.entity_embeddings_mat,
                            self.all_entity_sketches)
Wrapper to encode a single entity into its EmQL representation.
language/emql/model.py