@@ -111,18 +111,13 @@ class Embedder {
111
111
const Vocab &Vocabulary,
112
112
unsigned Dimension);
113
113
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.
117
115
const InstEmbeddingsMap &getInstVecMap () const { return InstVecMap; }
118
116
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.
122
118
const BBEmbeddingsMap &getBBVecMap () const { return BBVecMap; }
123
119
124
- // / Returns the vector representation for a given function corresponding to
125
- // / the IR2Vec algorithm.
120
+ // / Returns the embedding for the current function.
126
121
const Embedding &getFunctionVector () const { return FuncVector; }
127
122
};
128
123
@@ -131,15 +126,13 @@ class Embedder {
131
126
// / representations obtained from the Vocabulary.
132
127
class SymbolicEmbedder : public Embedder {
133
128
private:
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.
136
130
Embedding computeBB2Vec (const BasicBlock &BB);
137
131
138
- // / Utility function to compute the vector representation for a given type.
132
+ // / Utility function to compute the embedding for a given type.
139
133
Embedding getTypeEmbedding (const Type *Ty) const ;
140
134
141
- // / Utility function to compute the vector representation for a given
142
- // / operand.
135
+ // / Utility function to compute the embedding for a given operand.
143
136
Embedding getOperandEmbedding (const Value *Op) const ;
144
137
145
138
public:
0 commit comments