oarust 0.8.6

GPU-first semantic computing engine written in Rust
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29

// @generated by tool/gen/fn/generate.py; DO NOT EDIT.
// schemas=matrix_elemwise.json,matrix_blas.json,matrix_reduce.json,matrix_rng.json,matrix_index.json,matrix_view.json schema_version=1 generator_version=64
// schema_sha256=7b6f6f1da1e8c37efaa239e33a5b8f167641366950074285aaec120d979c658f
//! Schema-owned foundational attachment facade.

#[path = "matrix/elemwise.gen.rs"]
mod elemwise;
pub(in crate::matrix) use elemwise::{
	record_abs, record_add, record_clamp_max, record_clamp_min, record_copy, record_div, record_exp,
	record_log, record_mul, record_reciprocal, record_scale, record_sqrt, record_sub,
};
#[path = "matrix/blas.gen.rs"]
mod blas;
pub(in crate::matrix) use blas::record_mat_mul_nt;
#[path = "matrix/reduce.gen.rs"]
mod reduce;
pub(in crate::matrix) use reduce::{record_log_softmax, record_softmax, record_sum};
#[path = "matrix/rng.gen.rs"]
mod rng;
pub(in crate::matrix) use rng::record_dropout;
#[path = "matrix/index.gen.rs"]
mod index;
pub(in crate::matrix) use index::{
	record_concat, record_gather, record_gather_last_dim, record_repeat_interleave, record_slice,
	record_transpose,
};
#[path = "matrix/view.gen.rs"]
mod view;
pub(in crate::matrix) use view::record_reshape;