@@ -111,18 +111,13 @@ class Embedder {
111111 const Vocab &Vocabulary,
112112 unsigned Dimension);
113113
114- // / Returns a map containing instructions and the corresponding vector
115- // / representations for a given module corresponding to the IR2Vec
116- // / algorithm.
114+ // / Returns a map containing instructions and the corresponding embeddings.
117115 const InstEmbeddingsMap &getInstVecMap () const { return InstVecMap; }
118116
119- // / Returns a map containing basic block and the corresponding vector
120- // / representations for a given module corresponding to the IR2Vec
121- // / algorithm.
117+ // / Returns a map containing basic block and the corresponding embeddings.
122118 const BBEmbeddingsMap &getBBVecMap () const { return BBVecMap; }
123119
124- // / Returns the vector representation for a given function corresponding to
125- // / the IR2Vec algorithm.
120+ // / Returns the embedding for the current function.
126121 const Embedding &getFunctionVector () const { return FuncVector; }
127122};
128123
@@ -131,15 +126,13 @@ class Embedder {
131126// / representations obtained from the Vocabulary.
132127class SymbolicEmbedder : public Embedder {
133128private:
134- // / Utility function to compute the vector representation for a given basic
135- // / block.
129+ // / Utility function to compute the embedding for a given basic block.
136130 Embedding computeBB2Vec (const BasicBlock &BB);
137131
138- // / Utility function to compute the vector representation for a given type.
132+ // / Utility function to compute the embedding for a given type.
139133 Embedding getTypeEmbedding (const Type *Ty) const ;
140134
141- // / Utility function to compute the vector representation for a given
142- // / operand.
135+ // / Utility function to compute the embedding for a given operand.
143136 Embedding getOperandEmbedding (const Value *Op) const ;
144137
145138public:
0 commit comments