autograd

Functions

FunctionDescription
grad_add_a
grad_add_b
grad_sub_a
grad_sub_b
grad_mul_a
grad_mul_b
grad_div_a
grad_div_b
grad_relu
grad_exp
grad_log
grad_tanh
grad_tensor_add
grad_tensor_mul
grad_tensor_relu
grad_tensor_sum
grad_matmul
grad_check_close

Details

grad_add_a

fn grad_add_a(grad_c: i64) -> i64

grad_add_b

fn grad_add_b(grad_c: i64) -> i64

grad_sub_a

fn grad_sub_a(grad_c: i64) -> i64

grad_sub_b

fn grad_sub_b(grad_c: i64) -> i64

grad_mul_a

fn grad_mul_a(b: i64, grad_c: i64) -> i64

grad_mul_b

fn grad_mul_b(a: i64, grad_c: i64) -> i64

grad_div_a

fn grad_div_a(b: i64, grad_c: i64) -> i64

grad_div_b

fn grad_div_b(a: i64, b: i64, grad_c: i64) -> i64

grad_relu

fn grad_relu(a: i64, grad_c: i64) -> i64

grad_exp

fn grad_exp(c: i64, grad_c: i64) -> i64

grad_log

fn grad_log(a: i64, grad_c: i64) -> i64

grad_tanh

fn grad_tanh(c: i64, grad_c: i64) -> i64

grad_tensor_add

fn grad_tensor_add(grad_c: i64, grad_a: i64, grad_b: i64) -> i64

grad_tensor_mul

fn grad_tensor_mul(a: i64, b: i64, grad_c: i64, grad_a: i64, grad_b: i64) -> i64

grad_tensor_relu

fn grad_tensor_relu(a: i64, grad_c: i64, grad_a: i64) -> i64

grad_tensor_sum

fn grad_tensor_sum(grad_c_scalar: i64, grad_a: i64) -> i64

grad_matmul

fn grad_matmul(a: i64, b: i64, grad_c: i64, grad_a: i64, grad_b: i64) -> i64

grad_check_close

fn grad_check_close(analytical: i64, numerical: i64, eps: i64) -> i64