24 int generateEncodings(llvm::Module &M, IR2VecMode mode,
char level =
'\0',
25 std::string funcName =
"", std::ostream *o =
nullptr,
26 int cls = -1,
float WO = 1,
float WA = 0.2,
29 llvm::SmallMapVector<const llvm::Instruction *, Vector, 128> instVecMap;
30 llvm::SmallMapVector<const llvm::Function *, Vector, 16> funcVecMap;
35 Embeddings(llvm::Module &M, IR2VecMode mode, std::string funcName =
"",
36 float WO = 1,
float WA = 0.2,
float WT = 0.5) {
37 generateEncodings(M, mode,
'\0', funcName,
nullptr, -1, WO, WA, WT);
43 Embeddings(llvm::Module &M, IR2VecMode mode,
char level, std::ostream *o,
44 std::string funcName =
"",
float WO = 1,
float WA = 0.2,
46 generateEncodings(M, mode, level, funcName, o, -1, WO, WA, WT);
52 llvm::SmallMapVector<const llvm::Instruction *, Vector, 128> &
60 llvm::SmallMapVector<const llvm::Function *, Vector, 16> &
67 Vector &getProgramVector() {
return pgmVector; }