MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
Loading...
Searching...
No Matches
Public Member Functions | Static Public Member Functions | Private Member Functions | Private Attributes | List of all members
MLBridge::TensorflowSerDes Class Reference

TensorflowSerDes - Serialization/Deserialization to support TF AOT models. More...

#include <tensorflowSerDes.h>

Inheritance diagram for MLBridge::TensorflowSerDes:
Inheritance graph
[legend]
Collaboration diagram for MLBridge::TensorflowSerDes:
Collaboration graph
[legend]

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
 

Detailed Description

TensorflowSerDes - Serialization/Deserialization to support TF AOT models.

Definition at line 22 of file tensorflowSerDes.h.

Constructor & Destructor Documentation

◆ TensorflowSerDes()

MLBridge::TensorflowSerDes::TensorflowSerDes ( )
inline

Definition at line 24 of file tensorflowSerDes.h.

Member Function Documentation

◆ classof()

static bool MLBridge::TensorflowSerDes::classof ( const BaseSerDes * S)
inlinestatic

Definition at line 26 of file tensorflowSerDes.h.

◆ cleanDataStructures()

void MLBridge::TensorflowSerDes::cleanDataStructures ( )
inlineoverridevirtual

Implements MLBridge::BaseSerDes.

Definition at line 42 of file tensorflowSerDes.h.

◆ deserializeUntyped()

void * MLBridge::TensorflowSerDes::deserializeUntyped ( void * data)
inlineoverrideprivatevirtual

Implements MLBridge::BaseSerDes.

Definition at line 45 of file tensorflowSerDes.h.

◆ getSerializedData()

void * MLBridge::TensorflowSerDes::getSerializedData ( )
inlineoverridevirtual

Implements MLBridge::BaseSerDes.

Definition at line 41 of file tensorflowSerDes.h.

◆ setRequest()

void MLBridge::TensorflowSerDes::setRequest ( void * request)
inlineoverridevirtual

Reimplemented from MLBridge::BaseSerDes.

Definition at line 36 of file tensorflowSerDes.h.

Member Data Documentation

◆ CompiledModel

tensorflow::XlaCompiledCpuFunction* MLBridge::TensorflowSerDes::CompiledModel
private

Definition at line 46 of file tensorflowSerDes.h.


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