MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
|
Files | |
agent.h | |
Agent class to support ML/RL model inference via ONNX. | |
environment.h | |
Base Environment class to support ONNX based inference of RL models. | |
onnx.h | |
This file defines the ONNXModel class, which is a wrapper around the ONNX C++ interface. | |
ONNXModelRunner.h | |
ONNXModelRunner class supporting communication via ONNX C++ Runtime. | |
utils.h | |