Struct Cpu 

pub struct Cpu;

Trait Implementations

impl AbsOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl AcosOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl AddOps for Cpu

fn array_array<T: Num>( to: *mut T, lhs: *const T, rhs: *const T, num_elm: usize, to_stride: usize, lhs_stride: usize, rhs_stride: usize, )

fn array_assign<T: Num>( to: *mut T, rhs: *const T, num_elm: usize, to_stride: usize, rhs_stride: usize, )

fn scalar<T: Num>( to: *mut T, lhs: *const T, rhs: T, num_elm: usize, to_stride: usize, lhs_stride: usize, )

fn scalar_assign<T: Num>(to: *mut T, rhs: T, num_elm: usize, to_stride: usize)

fn scalar_ptr<T: Num>( to: *mut T, lhs: *const T, scalar: *const T, to_stride: usize, lhs_stride: usize, num_elm: usize, )

fn scalar_assign_ptr<T: Num>( to: *mut T, scalar: *const T, num_elm: usize, to_stride: usize, )

impl AsinOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl Asum for Cpu

fn asum<T: Num>(n: usize, x: *const T, incx: usize) -> T

impl AtanOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl BatchNormalization for Cpu

fn batch_norm_2d_forward_train<T: Num>( momentum: f64, x: Matrix<Ref<&T>, DimDyn, Self>, y: Matrix<Ref<&mut T>, DimDyn, Self>, scale: Matrix<Ref<&T>, DimDyn, Self>, bias: Matrix<Ref<&T>, DimDyn, Self>, mean: Matrix<Ref<&mut T>, DimDyn, Self>, variance: Matrix<Ref<&mut T>, DimDyn, Self>, saving_mean: Option<Matrix<Ref<&mut T>, DimDyn, Self>>, saving_inv_variance: Option<Matrix<Ref<&mut T>, DimDyn, Self>>, _: &Option<BatchNorm2dConfig<T>>, )

fn batch_norm_2d_backward<T: Num>( x: Matrix<Ref<&T>, DimDyn, Self>, y_grad: Matrix<Ref<&T>, DimDyn, Self>, x_grad: Matrix<Ref<&mut T>, DimDyn, Self>, scale: Matrix<Ref<&T>, DimDyn, Self>, scale_grad: Matrix<Ref<&mut T>, DimDyn, Self>, bias_grad: Matrix<Ref<&mut T>, DimDyn, Self>, saving_mean: Option<Matrix<Ref<&T>, DimDyn, Self>>, saving_inv_variance: Option<Matrix<Ref<&T>, DimDyn, Self>>, _: &Option<BatchNorm2dBackwardConfig<T>>, )

fn bach_norm_2d_forward_inference<T: Num>( x: Matrix<Ref<&T>, DimDyn, Self>, y: Matrix<Ref<&mut T>, DimDyn, Self>, scale: Matrix<Ref<&T>, DimDyn, Self>, bias: Matrix<Ref<&T>, DimDyn, Self>, mean: Matrix<Ref<&T>, DimDyn, Self>, variance: Matrix<Ref<&T>, DimDyn, Self>, _: &Option<BatchNorm2dInferenceConfig<T>>, )

impl ClipOps for Cpu

fn clip<T: Num>( input: *const T, output: *mut T, size: usize, stride_in: usize, stride_out: usize, min: T, max: T, )

fn clip_assign<T: Num>( input: *mut T, size: usize, stride: usize, min: T, max: T, )

fn clip_backward<T: Num>( input: *const T, mask: *mut T, max: T, min: T, size: usize, stride_in: usize, stride_out: usize, )

fn clip_backward_assign<T: Num>( mask: *mut T, max: T, min: T, size: usize, stride: usize, )

impl Clone for Cpu

fn clone(&self) -> Cpu

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Conv2d for Cpu

fn conv2d<T: Num>( input: Matrix<Ref<&T>, DimDyn, Self>, y: Matrix<Ref<&mut T>, DimDyn, Self>, filter: Matrix<Ref<&T>, DimDyn, Self>, pad_h: usize, pad_w: usize, stride_h: usize, stride_w: usize, dilation_h: usize, dilation_w: usize, _config: Option<&Conv2dConfig<T>>, )

fn conv2d_bckwd_data<T: Num>( dy: Matrix<Ref<&T>, DimDyn, Self>, dx: Matrix<Ref<&mut T>, DimDyn, Self>, filter: Matrix<Ref<&T>, DimDyn, Self>, pad_h: usize, pad_w: usize, stride_h: usize, stride_w: usize, dilation_h: usize, dilation_w: usize, _config: Option<&Conv2dConfig<T>>, )

fn conv2d_bckwd_filter<T: Num>( input: Matrix<Ref<&T>, DimDyn, Self>, dy: Matrix<Ref<&T>, DimDyn, Self>, df: Matrix<Ref<&mut T>, DimDyn, Self>, pad_h: usize, pad_w: usize, stride_h: usize, stride_w: usize, dilation_h: usize, dilation_w: usize, _config: Option<&Conv2dConfig<T>>, )

fn conv2d_forward_bias<T: Num>( input: Matrix<Ref<&T>, DimDyn, Self>, y: Matrix<Ref<&mut T>, DimDyn, Self>, bias: Matrix<Ref<&T>, DimDyn, Self>, )

fn conv2d_bckwd_bias<T: Num>( dy: Matrix<Ref<&T>, DimDyn, Self>, dx: Matrix<Ref<&mut T>, DimDyn, Self>, )

impl CopyBlas for Cpu

fn copy_raw<T: Num>(n: usize, x: *const T, incx: usize, y: *mut T, incy: usize)

impl CosOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl CoshOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl Debug for Cpu

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl Default for Cpu

fn default() -> Cpu

Returns the “default value” for a type.

impl<'de> Deserialize<'de> for Cpu

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl DeviceBase for Cpu

fn raw_drop_ptr<T>(ptr: *mut T)

fn mem_pool_drop_ptr(ptr: *mut u8) -> Result<(), MemPoolError>

fn clone_ptr<T>(ptr: *const T, len: usize) -> *mut T

fn assign_item<T>(ptr: *mut T, offset: usize, value: T)

fn get_item<T>(ptr: *const T, offset: usize) -> T

fn from_vec<T>(vec: Vec<T>) -> *mut T

fn zeros<T: Num>(len: usize) -> *mut T

fn raw_alloc(num_bytes: usize) -> Result<*mut u8, String>

fn mem_pool_alloc(num_bytes: usize) -> Result<*mut u8, MemPoolError>

fn drop_ptr<T>(ptr: *mut T)

fn alloc(num_bytes: usize) -> Result<*mut u8, MemPoolError>

impl DivOps for Cpu

fn array_array<T: Num>( to: *mut T, lhs: *const T, rhs: *const T, num_elm: usize, to_stride: usize, lhs_stride: usize, rhs_stride: usize, )

fn array_assign<T: Num>( to: *mut T, rhs: *const T, num_elm: usize, to_stride: usize, rhs_stride: usize, )

fn scalar<T: Num>( to: *mut T, lhs: *const T, rhs: T, num_elm: usize, to_stride: usize, lhs_stride: usize, )

fn scalar_assign<T: Num>(to: *mut T, rhs: T, num_elm: usize, to_stride: usize)

fn scalar_ptr<T: Num>( to: *mut T, lhs: *const T, scalar: *const T, to_stride: usize, lhs_stride: usize, num_elm: usize, )

fn scalar_assign_ptr<T: Num>( to: *mut T, scalar: *const T, num_elm: usize, to_stride: usize, )

impl Dropout for Cpu

fn dropout<T: Num>( x: &Matrix<Ref<&T>, DimDyn, Self>, state: &mut DropoutState<T, Self>, ) -> Matrix<Owned<T>, DimDyn, Self>

fn dropout_grad<T: Num>( dy: &Matrix<Ref<&T>, DimDyn, Self>, state: &DropoutState<T, Self>, ) -> Matrix<Owned<T>, DimDyn, Self>

impl ExpOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl Gemm for Cpu

fn gemm_unchecked<T: Num>( transa: BlasTrans, transb: BlasTrans, m: usize, n: usize, k: usize, alpha: T, a: *const T, lda: usize, b: *const T, ldb: usize, beta: T, c: *mut T, ldc: usize, )

impl LogOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl MaxIdx for Cpu

fn max_idx<T: Num>(input: *const T, size: usize, stride: usize) -> usize

impl MulOps for Cpu

fn array_array<T: Num>( to: *mut T, lhs: *const T, rhs: *const T, num_elm: usize, to_stride: usize, lhs_stride: usize, rhs_stride: usize, )

fn array_assign<T: Num>( to: *mut T, rhs: *const T, num_elm: usize, to_stride: usize, rhs_stride: usize, )

fn scalar<T: Num>( to: *mut T, lhs: *const T, rhs: T, num_elm: usize, to_stride: usize, lhs_stride: usize, )

fn scalar_assign<T: Num>(to: *mut T, rhs: T, num_elm: usize, to_stride: usize)

fn scalar_ptr<T: Num>( to: *mut T, lhs: *const T, scalar: *const T, to_stride: usize, lhs_stride: usize, num_elm: usize, )

fn scalar_assign_ptr<T: Num>( to: *mut T, scalar: *const T, num_elm: usize, to_stride: usize, )

impl Pool2dImpl for Cpu

fn pool2d<T: Num>( input: Matrix<Ref<&T>, DimDyn, Self>, output: Matrix<Ref<&mut T>, DimDyn, Self>, kernel: (usize, usize), stride: (usize, usize), padding: (usize, usize), _config: &Pool2dConfig<T>, ) -> Result<(), String>

fn pool2d_backward<T: Num>( input: Matrix<Ref<&T>, DimDyn, Self>, input_grad: Matrix<Ref<&mut T>, DimDyn, Self>, _output: Matrix<Ref<&T>, DimDyn, Self>, output_grad: Matrix<Ref<&T>, DimDyn, Self>, kernel_shape: (usize, usize), stride: (usize, usize), padding: (usize, usize), _config: &Pool2dConfig<T>, )

impl PowOws for Cpu

fn array<T: Num>( to: *mut T, other: *const T, scalar: T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, scalar: T, num_elm: usize, to_stride: usize)

impl ReluOps for Cpu

fn relu<T: Num>( input: *const T, output: *mut T, alpha: T, size: usize, input_stride: usize, output_stride: usize, )

fn relu_backward_mask<T: Num>( input: *const T, mask: *mut T, alpha: T, size: usize, input_stride: usize, mask_stride: usize, )

impl Serialize for Cpu

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.

impl SinOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl SinhOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl SqrtOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl SubOps for Cpu

fn array_array<T: Num>( to: *mut T, lhs: *const T, rhs: *const T, num_elm: usize, to_stride: usize, lhs_stride: usize, rhs_stride: usize, )

fn array_assign<T: Num>( to: *mut T, rhs: *const T, num_elm: usize, to_stride: usize, rhs_stride: usize, )

fn scalar<T: Num>( to: *mut T, lhs: *const T, rhs: T, num_elm: usize, to_stride: usize, lhs_stride: usize, )

fn scalar_assign<T: Num>(to: *mut T, rhs: T, num_elm: usize, to_stride: usize)

fn scalar_ptr<T: Num>( to: *mut T, lhs: *const T, scalar: *const T, to_stride: usize, lhs_stride: usize, num_elm: usize, )

fn scalar_assign_ptr<T: Num>( to: *mut T, scalar: *const T, num_elm: usize, to_stride: usize, )

impl TanOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl TanhOps for Cpu

fn array<T: Num>( to: *mut T, other: *const T, num_elm: usize, to_stride: usize, other_stride: usize, )

fn array_assign<T: Num>(to: *mut T, num_elm: usize, to_stride: usize)

impl Copy for Cpu

impl Device for Cpu