MLCompilerBridge
Tools for streamlining communication with ML models for compiler optimizations.
Loading...
Searching...
No Matches
ONNXModelRunner Directory Reference
Directory dependency graph for ONNXModelRunner:
include/MLModelRunner/ONNXModelRunner

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