llvm::EmbeddedModelRunnerOptions Struct Reference

ReleaseModeModelRunner - production mode implementation of the MLModelRunner. More...

#include "llvm/Analysis/ReleaseModeModelRunner.h"

Public Member Functions
EmbeddedModelRunnerOptions &	setFeedPrefix (StringRef Value)
EmbeddedModelRunnerOptions &	setFetchPrefix (StringRef Value)
EmbeddedModelRunnerOptions &	setModelSelector (StringRef Value)

Public Attributes
StringRef	FeedPrefix = "feed_"
	Feed and Fetch feature prefixes - i.e.
StringRef	FetchPrefix = "fetch_"
StringRef	ModelSelector = ""
	ModelSelector is the name (recognized by the AOT-ed model) of a sub-model to use.

Detailed Description

ReleaseModeModelRunner - production mode implementation of the MLModelRunner.

It uses an AOT-compiled SavedModel for efficient execution.

Definition at line 29 of file ReleaseModeModelRunner.h.

Member Function Documentation

setFeedPrefix()

EmbeddedModelRunnerOptions & llvm::EmbeddedModelRunnerOptions::setFeedPrefix ( StringRef Value )

inline

Definition at line 40 of file ReleaseModeModelRunner.h.

References FeedPrefix.

setFetchPrefix()

EmbeddedModelRunnerOptions & llvm::EmbeddedModelRunnerOptions::setFetchPrefix ( StringRef Value )

inline

Definition at line 44 of file ReleaseModeModelRunner.h.

References FetchPrefix.

setModelSelector()

EmbeddedModelRunnerOptions & llvm::EmbeddedModelRunnerOptions::setModelSelector ( StringRef Value )

inline

Definition at line 48 of file ReleaseModeModelRunner.h.

References ModelSelector.

Member Data Documentation

FeedPrefix

StringRef llvm::EmbeddedModelRunnerOptions::FeedPrefix = "feed_"

Feed and Fetch feature prefixes - i.e.

a feature named "foo" will be looked up as {FeedPrefix}_foo; and the output named "bar" will be looked up as {FetchPrefix}_bar

Definition at line 33 of file ReleaseModeModelRunner.h.

Referenced by setFeedPrefix().

FetchPrefix

StringRef llvm::EmbeddedModelRunnerOptions::FetchPrefix = "fetch_"

Definition at line 34 of file ReleaseModeModelRunner.h.

Referenced by setFetchPrefix().

ModelSelector

StringRef llvm::EmbeddedModelRunnerOptions::ModelSelector = ""

ModelSelector is the name (recognized by the AOT-ed model) of a sub-model to use.

"" is allowed if the model doesn't support sub-models.

Definition at line 38 of file ReleaseModeModelRunner.h.

Referenced by setModelSelector().

---

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

• include/llvm/Analysis/ReleaseModeModelRunner.h