This file implements a model runner wrapping an AOT compiled ML model. More...
#include "MLModelRunner/MLModelRunner.h"#include "SerDes/TensorSpec.h"#include <memory>#include <vector>
Include dependency graph for TFModelRunner.h:
Go to the source code of this file.
Classes | |
| class | MLBridge::TFModelRunner< TGen > |
| TFModelRunner - TF Compiled model implementation of the MLModelRunner. More... | |
| class | MLBridge::NoopSavedModelImpl |
A mock class satisfying the interface expected by ReleaseModeModelRunner for its TGen parameter. More... | |
Namespaces | |
| namespace | MLBridge |
Macros | |
| #define | NOOP_MODEL_ERRMSG |
Functions | |
| template<class T > | |
| bool | MLBridge::isEmbeddedModelEvaluatorValid () |
| template<> | |
| bool | MLBridge::isEmbeddedModelEvaluatorValid< NoopSavedModelImpl > () |
Detailed Description
This file implements a model runner wrapping an AOT compiled ML model.
Only inference is supported.
Definition in file TFModelRunner.h.
Macro Definition Documentation
◆ NOOP_MODEL_ERRMSG
| #define NOOP_MODEL_ERRMSG |
Value:
"The mock AOT-ed saved model is a compile-time stub and should not be " \
"called."
Definition at line 86 of file TFModelRunner.h.
