tensorrt-rs 0.3.0

Rust library for using Nvidia's TensorRT deep learning acceleration library
1
2
3
4
5
6
7
8
9
10
11

use tensorrt_sys::destroy_network;

pub struct Network {
    pub(crate) internal_network: *mut tensorrt_sys::Network_t,
}

impl Drop for Network {
    fn drop(&mut self) {
        unsafe { destroy_network(self.internal_network) };
    }
}