Class for computing the Flow-aware embeddings of IR2Vec. More...

Inheritance diagram for llvm::ir2vec::FlowAwareEmbedder:

Public Member Functions
	FlowAwareEmbedder (const Function &F, const Vocabulary &Vocab)

Public Member Functions inherited from llvm::ir2vec::Embedder
virtual	~Embedder ()=default

LLVM_ABI const InstEmbeddingsMap &	getInstVecMap () const
	Returns a map containing instructions and the corresponding embeddings for the function F if it has been computed.

LLVM_ABI const BBEmbeddingsMap &	getBBVecMap () const
	Returns a map containing basic block and the corresponding embeddings for the function F if it has been computed.

LLVM_ABI const Embedding &	getBBVector (const BasicBlock &BB) const
	Returns the embedding for a given basic block in the function F if it has been computed.

LLVM_ABI const Embedding &	getFunctionVector () const
	Computes and returns the embedding for the current function.

Additional Inherited Members
Static Public Member Functions inherited from llvm::ir2vec::Embedder
static LLVM_ABI std::unique_ptr< Embedder >	create (IR2VecKind Mode, const Function &F, const Vocabulary &Vocab)
	Factory method to create an Embedder object.

Protected Member Functions inherited from llvm::ir2vec::Embedder
LLVM_ABI	Embedder (const Function &F, const Vocabulary &Vocab)

void	computeEmbeddings () const
	Function to compute embeddings.

virtual void	computeEmbeddings (const BasicBlock &BB) const =0
	Function to compute the embedding for a given basic block.

Protected Attributes inherited from llvm::ir2vec::Embedder
const Function &	F

const Vocabulary &	Vocab

const unsigned	Dimension
	Dimension of the vector representation; captured from the input vocabulary.

const float	OpcWeight
	Weights for different entities (like opcode, arguments, types) in the IR instructions to generate the vector representation.

const float	TypeWeight

const float	ArgWeight

Embedding	FuncVector

BBEmbeddingsMap	BBVecMap

InstEmbeddingsMap	InstVecMap

Detailed Description

Class for computing the Flow-aware embeddings of IR2Vec.

Flow-aware embeddings build on the vocabulary, just like Symbolic embeddings, and additionally capture the flow information in the IR.

Definition at line 401 of file IR2Vec.h.

Constructor & Destructor Documentation

llvm::ir2vec::FlowAwareEmbedder::FlowAwareEmbedder	(	const Function &	F,
		const Vocabulary &	Vocab
	)

inline

Definition at line 406 of file IR2Vec.h.

The documentation for this class was generated from the following files: