LLVM: llvm::MLModelRunner Class Reference (original) (raw)

MLModelRunner interface: abstraction of a mechanism for evaluating a ML model. More...

#include "[llvm/Analysis/MLModelRunner.h](MLModelRunner%5F8h%5Fsource.html)"

Public Types
enum class Kind : int { Unknown, Release, Development, NoOp, Interactive }
Public Member Functions
MLModelRunner (const MLModelRunner &)=delete
MLModelRunner & operator= (const MLModelRunner &)=delete
virtual ~MLModelRunner ()=default
template<typename T>
T evaluate ()
template<typename T, typename I>
T * getTensor (I FeatureID)
template<typename T, typename I>
const T * getTensor (I FeatureID) const
void * getTensorUntyped (size_t Index)
const void * getTensorUntyped (size_t Index) const
Kind getKind () const
virtual void switchContext (StringRef Name)

MLModelRunner interface: abstraction of a mechanism for evaluating a ML model.

More abstractly, evaluating a function that has as tensors as arguments, described via TensorSpecs, and returns a tensor. Currently, the latter is assumed to be a scalar, in absence of more elaborate scenarios. NOTE: feature indices are expected to be consistent all accross MLModelRunners (pertaining to the same model), and also Loggers (see TFUtils.h)

Definition at line 26 of file MLModelRunner.h.

◆ Kind

enum class llvm::MLModelRunner::Kind : int strong
Enumerator
Unknown
Release
Development
NoOp
Interactive

Definition at line 52 of file MLModelRunner.h.

◆ ~MLModelRunner()

virtual llvm::MLModelRunner::~MLModelRunner ( ) virtualdefault

◆ MLModelRunner() [2/2]

llvm::MLModelRunner::MLModelRunner ( LLVMContext & Ctx, Kind Type, size_t NumInputs ) inlineprotected

◆ evaluate()

template<typename T>

T llvm::MLModelRunner::evaluate ( ) inline

◆ evaluateUntyped()

virtual void * llvm::MLModelRunner::evaluateUntyped ( ) protectedpure virtual

◆ getKind()

Kind llvm::MLModelRunner::getKind ( ) const inline

◆ getTensor() [1/2]

template<typename T, typename I>

T * llvm::MLModelRunner::getTensor ( I FeatureID) inline

◆ getTensor() [2/2]

template<typename T, typename I>

const T * llvm::MLModelRunner::getTensor ( I FeatureID) const inline

◆ getTensorUntyped() [1/2]

void * llvm::MLModelRunner::getTensorUntyped ( size_t Index) inline

◆ getTensorUntyped() [2/2]

const void * llvm::MLModelRunner::getTensorUntyped ( size_t Index) const inline

◆ operator=()

References MLModelRunner().

◆ setUpBufferForTensor()

void llvm::MLModelRunner::setUpBufferForTensor ( size_t Index, const TensorSpec & Spec, void * Buffer ) inlineprotected

◆ switchContext()

virtual void llvm::MLModelRunner::switchContext ( StringRef Name) inlinevirtual

◆ Ctx

◆ Type

const Kind llvm::MLModelRunner::Type protected

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