|
| | NeuralError.h |
| | Defines error handling classes for the neural network module.
|
| |
| | NeuralNet.h |
| | Contains the NeuralNet interface for neural network inference.
|
| |
| | NeuralNetFactory.h |
| | Factory class for creating NeuralNet instances.
|
| |
| | NeuralUtil.h |
| |
| | OrtNeuralNet.h |
| | Header file for the OrtNeuralNet class, which implements the NeuralNet interface using ONNX Runtime. It also handles the conversion between Tensor and ONNX Runtime's Ort::Value.
|
| |
| | ReduceOps.h |
| | Contains reduction operations for tensors such as sum, min, max, argmin, and argmax.
|
| |
| | Tensor.h |
| | Contains the Tensor class for representing N-dimensional arrays with various data types.
|
| |