def simple_follow_op(self, params, ent_embs, rel_embs,
                       ent_sketch, rel_sketch):
    """Wrapper of follow operation inside EmQL."""
    if not params['use_cm_sketch']:
      ent_sketch, rel_sketch = None, None

    object_embs, object_sketch, unused_topk_fact_logits = module.follow_op(
        params, ent_embs, ent_sketch, rel_embs, rel_sketch,
        self.all_fact_subjids, self.all_fact_relids, self.all_fact_objids,
        self.entity_embeddings_mat, self.kb_embeddings_mat,
        self.all_entity_sketches, self.all_relation_sketches)
    return object_embs, object_sketch
Wrapper of follow operation inside EmQL.
language/emql/model.py