ndrs 0.5.0

A tensor library with GPU support
Documentation
1
2
3
4
5
6
7
8
9
10
11
mod device;
mod elementwise_kernel;
mod event;
mod raw_kernel;
mod stream;

pub use device::{get_device, get_device_context, get_device_count, is_available, set_device};
pub use elementwise_kernel::ElementwiseKernel;
pub use event::Event;
pub use raw_kernel::RawKernel;
pub use stream::{Stream, default_stream, get_stream, set_stream};