MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
|
Supporting new SerDes: More...
#include "MLModelRunner/Utils/Debug.h"
#include "google/protobuf/extension_set.h"
#include "google/protobuf/message.h"
#include "llvm/Support/raw_ostream.h"
#include <cassert>
#include <map>
#include <string>
#include <vector>
Go to the source code of this file.
Classes | |
class | MLBridge::BaseSerDes |
Namespaces | |
namespace | MLBridge |
Macros | |
#define | SUPPORTED_TYPES(M) |
#define | SET_FEATURE(TYPE, _) |
setFeature() is used to set the features of the data structure used for communication. | |
Enumerations | |
enum class | MLBridge::SerDesKind : int { MLBridge::Unknown , MLBridge::Json , MLBridge::Bitstream , MLBridge::Protobuf , MLBridge::Tensorflow } |
This is the base class for SerDes. More... | |
Supporting new SerDes:
Definition in file baseSerDes.h.
#define SET_FEATURE | ( | TYPE, | |
_ ) |
setFeature() is used to set the features of the data structure used for communication.
The features are set as key-value pairs. The key is a string and the value can be any of the supported types.
Definition at line 54 of file baseSerDes.h.
#define SUPPORTED_TYPES | ( | M | ) |
Definition at line 32 of file baseSerDes.h.