ONNX networks¶
-
class OnnxNetwork : public virtual klartraum::ComputeGraphElement, public virtual klartraum::ComputeGraphGroup¶
Public Functions
-
OnnxNetwork(VulkanContext &vulkanContext, const std::string &modelPath)¶
ONNX Neural Network Graph Representation.
This class provides GPU-accelerated execution of ONNX models using Klartraum ComputeGraph Framework. It supports:
Loading and parsing ONNX model files
Converting ONNX operations to ComputeGraphElement instances
Managing GPU memory for tensors and intermediate results by ComputeGraphElement instances
-
OnnxNetwork(VulkanContext &vulkanContext, const std::string &modelPath)¶