MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
Loading...
Searching...
No Matches
Class Hierarchy

Go to the graphical class hierarchy

This inheritance list is sorted roughly, but not completely, alphabetically:
[detail level 123]
 CMLBridge::AgentAgent is a wrapper around the ONNXModel class, interfaces with the Environment class to support ML/RL model inference via ONNXModel
 CMLBridge::BaseSerDes
 CMLBridge::BitstreamSerDesBitstreamSerDes - Bitstream Serialization/Deserialization which sends header information followed by the raw data
 CMLBridge::JsonSerDesJsonSerDes - Json Serialization/Deserialization using LLVM's json library
 CMLBridge::ProtobufSerDesProtobufSerDes - Protobuf Serialization/Deserialization to support gRPC communication
 CMLBridge::TensorflowSerDesTensorflowSerDes - Serialization/Deserialization to support TF AOT models
 CEnvironment
 CMLBridge::Environment
 CMLBridge::MLModelRunnerMLModelRunner - The main interface for interacting with the ML models
 CMLBridge::ONNXModelRunnerONNXModelRunner is the main user facing class that interfaces with the Environment and Agent classes to support ML/RL model inference via ONNXModel
 CMLBridge::PipeModelRunnerA MLModelRunner that asks for advice from an external agent, or host
 CMLBridge::TFModelRunner< TGen >TFModelRunner - TF Compiled model implementation of the MLModelRunner
 CMLBridge::gRPCModelRunner< Client, Stub, Request, Response >This class is used to create the grpc model runner object
 CMLBridge::NoopSavedModelImplA mock class satisfying the interface expected by ReleaseModeModelRunner for its TGen parameter
 CONNXModel
 CONNXModelRunner
 CPipeModelRunnerWrapper
 Ccompilerinterface.SerDes.SerDesClass for serialization and deserialization in various formats for communication
 Ccompilerinterface.log_reader.TensorSpec
 CMLBridge::TensorSpec
 Ccompilerinterface.log_reader.TensorValue
 CABC
 Ccompilerinterface.BaseCompilerInterface.BaseCompilerInterfaceThis base class specifies methods for communication with compiler
 Ccompilerinterface.GrpcCompilerInterface.GrpcCompilerInterfaceThis class implements methods for communication with compiler using gRPC
 Ccompilerinterface.PipeCompilerInterface.PipeCompilerInterfaceThis class implements methods for communication with compiler using pipes