MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
|
TensorflowSerDes - Serialization/Deserialization to support TF AOT models. More...
#include <tensorflowSerDes.h>
Public Member Functions | |
TensorflowSerDes () | |
void | setRequest (void *request) override |
void * | getSerializedData () override |
void | cleanDataStructures () override |
Public Member Functions inherited from MLBridge::BaseSerDes | |
SerDesKind | getKind () const |
virtual void | setFeature (const std::string &name, const google::protobuf::Message *value) |
virtual void | setFeature (const std::string &name, const std::vector< google::protobuf::Message * > &value) |
virtual void | setResponse (void *Response) |
size_t | getMessageLength () |
virtual void * | getRequest () |
virtual void * | getResponse () |
Static Public Member Functions | |
static bool | classof (const BaseSerDes *S) |
Private Member Functions | |
void * | deserializeUntyped (void *data) override |
Private Attributes | |
tensorflow::XlaCompiledCpuFunction * | CompiledModel |
Additional Inherited Members | |
Protected Member Functions inherited from MLBridge::BaseSerDes | |
BaseSerDes (SerDesKind Type) | |
Protected Attributes inherited from MLBridge::BaseSerDes | |
const SerDesKind | Type |
void * | RequestVoid |
void * | ResponseVoid |
size_t | MessageLength |
TensorflowSerDes - Serialization/Deserialization to support TF AOT models.
Definition at line 22 of file tensorflowSerDes.h.
|
inline |
Definition at line 24 of file tensorflowSerDes.h.
|
inlinestatic |
Definition at line 26 of file tensorflowSerDes.h.
|
inlineoverridevirtual |
Implements MLBridge::BaseSerDes.
Definition at line 42 of file tensorflowSerDes.h.
|
inlineoverrideprivatevirtual |
Implements MLBridge::BaseSerDes.
Definition at line 45 of file tensorflowSerDes.h.
|
inlineoverridevirtual |
Implements MLBridge::BaseSerDes.
Definition at line 41 of file tensorflowSerDes.h.
|
inlineoverridevirtual |
Reimplemented from MLBridge::BaseSerDes.
Definition at line 36 of file tensorflowSerDes.h.
|
private |
Definition at line 46 of file tensorflowSerDes.h.