nn
Functions
Details
linear_forward
fn linear_forward(x: i64, w: i64, bias: i64, out: i64) -> i64
backward_weights
fn backward_weights(x: i64, w: i64, grad_out: i64, grad_w: i64) -> i64
backward_bias
fn backward_bias(grad_out: i64, grad_b: i64, batch: i64, out_f: i64) -> i64
fn backward_input(w: i64, grad_out: i64, grad_in: i64) -> i64
relu_forward
fn relu_forward(a: i64, out: i64) -> i64
relu_backward
fn relu_backward(pre_relu: i64, grad_out: i64, grad_in: i64) -> i64
mse_loss
fn mse_loss(pred: i64, target: i64) -> i64
mse_backward
fn mse_backward(pred: i64, target: i64, grad_out: i64) -> i64
sgd_step
fn sgd_step(params: i64, grads: i64, lr: i64) -> i64
zero_tensor
fn zero_tensor(t: i64) -> i64
