Crate custos_math

source · [−]

Re-exports

pub use cpu::*;
pub use cuda::*;
pub use opencl::*;

Modules

cpu
cuda
nn
opencl

Structs

Matrix

A matrix using Buffer described with rows and columns

Traits

AdditionalOps
AssignOps

Assignment operations

BaseOps

Element-wise +, -, *, / operations for matrices.

ClipOp
ColOp
CudaTranspose
DiagflatOp
FnsOps
Gemm

Matrix multiplication. Uses provided device.

MaxOps
RandBuf
RandOp
RowOp
SumOps
TransposeOp

Functions

cl_diagflat
cl_to_cpu_lr

OpenCL

cl_to_cpu_s

OpenCL

cl_transpose
cu_clip
diagflat
ew_op
rand_slice
slice_transpose