|
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.