pulsatrix::CUDABackend Member List

This is the complete list of members for pulsatrix::CUDABackend, including all inherited members.

accumulate_rows(const float *in, float *out, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
adam_step(float *param, const float *grad, float *m, float *v, size_t n, float lr, float beta1, float beta2, float eps, float bias_correction1, float bias_correction2) override	pulsatrix::CUDABackend	virtual
add(const float *a, const float *b, float *out, size_t n) override	pulsatrix::CUDABackend	virtual
add_channel_vector(const float *in, const float *vec, float *out, size_t n, size_t c, size_t inner) override	pulsatrix::CUDABackend	virtual
add_row_vector(const float *in, const float *row, float *out, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
aggregator_backward(const float *x, const float *mean_pow, const float *grad_out, float *grad_in, size_t n, size_t cols, float p) override	pulsatrix::CUDABackend	virtual
aggregator_forward(const float *x, float *mean_pow, float *out, size_t n, size_t cols, float p) override	pulsatrix::CUDABackend	virtual
aggregator_lrp(const float *x, const float *mean_pow, const float *r_out, float *r_in, size_t n, size_t cols, float p, float eps) override	pulsatrix::CUDABackend	virtual
allocate(size_t bytes) override	pulsatrix::CUDABackend	virtual
avg_pool_backward(const float *grad_out, float *grad_in, size_t planes, size_t h, size_t w, size_t kh, size_t kw) override	pulsatrix::CUDABackend	virtual
avg_pool_forward(const float *in, float *out, size_t planes, size_t h, size_t w, size_t kh, size_t kw) override	pulsatrix::CUDABackend	virtual
axpby(float alpha, const float *x, float beta, const float *y, float *out, size_t n) override	pulsatrix::CUDABackend	virtual
batch_norm_backward(const float *grad_out, const float *gamma, const float *xhat, const float *channel_std, float *grad_in, float *gamma_grad, float *beta_grad, size_t n, size_t c, size_t spatial) override	pulsatrix::CUDABackend	virtual
batch_norm_eval_backward(const float *grad_out, const float *gamma, const float *xhat, const float *channel_std, float *grad_in, float *gamma_grad, float *beta_grad, size_t n, size_t c, size_t spatial) override	pulsatrix::CUDABackend	virtual
batch_norm_eval_forward(const float *in, const float *gamma, const float *beta, const float *running_mean, const float *running_var, float *xhat, float *out, float *channel_std, size_t n, size_t c, size_t spatial, float eps) override	pulsatrix::CUDABackend	virtual
batch_norm_forward(const float *in, const float *gamma, const float *beta, float *xhat, float *out, float *channel_std, size_t n, size_t c, size_t spatial, float eps) override	pulsatrix::CUDABackend	virtual
batch_norm_update_running(const float *in, float *running_mean, float *running_var, size_t n, size_t c, size_t spatial, float momentum) override	pulsatrix::CUDABackend	virtual
bce_with_logits(const float *logits, const float *target, float *out, size_t n) override	pulsatrix::CUDABackend	virtual
bce_with_logits_grad(const float *logits, const float *target, float *grad, size_t n, float scale) override	pulsatrix::CUDABackend	virtual
col2im_add(const float *col, float *out, size_t n, size_t c, size_t h, size_t w, const ConvGeometry &geometry) override	pulsatrix::CUDABackend	virtual
column_sums(const float *in, float *out, size_t rows, size_t cols, float beta) override	pulsatrix::CUDABackend	virtual
copy(void *dst, const void *src, size_t bytes, CopyDirection dir) override	pulsatrix::CUDABackend	virtual
copy_2d(float *dst, size_t dst_stride, const float *src, size_t src_stride, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
CUDABackend()	pulsatrix::CUDABackend
CUDABackend(const CUDABackend &)=delete	pulsatrix::CUDABackend
device() const noexcept override	pulsatrix::CUDABackend	inlinevirtual
dot(const float *a, const float *b, size_t n) override	pulsatrix::CUDABackend	virtual
dropout_forward(const float *in, float *out, float *mask, size_t n, float p, float scale, uint64_t seed, uint64_t offset) override	pulsatrix::CUDABackend	virtual
elementwise(ElementwiseOp op, const float *in, float *out, size_t n) override	pulsatrix::CUDABackend	virtual
elementwise_backward(ElementwiseOp op, const float *x, const float *grad_out, float *grad_in, size_t n) override	pulsatrix::CUDABackend	virtual
fill(void *ptr, float value, size_t n) override	pulsatrix::CUDABackend	virtual
free(void *ptr) noexcept override	pulsatrix::CUDABackend	virtual
gather_rows(const float *table, const float *indices, float *out, size_t count, size_t dim) override	pulsatrix::CUDABackend	virtual
gemm(const float *a, const float *b, float *out, size_t m, size_t k, size_t n) override	pulsatrix::CUDABackend	virtual
gemm_ex(const float *a, bool transpose_a, const float *b, bool transpose_b, float *out, size_t m, size_t k, size_t n, float beta) override	pulsatrix::CUDABackend	virtual
group_norm_backward(const float *grad_out, const float *gamma, const float *xhat, const float *group_std, float *grad_in, float *gamma_grad, float *beta_grad, size_t n, size_t c, size_t spatial, size_t num_groups) override	pulsatrix::CUDABackend	virtual
group_norm_forward(const float *in, const float *gamma, const float *beta, float *xhat, float *out, float *group_std, size_t n, size_t c, size_t spatial, size_t num_groups, float eps) override	pulsatrix::CUDABackend	virtual
gru_lrp_hprev(const float *h_prev, const float *w_hn, const float *hn, const float *r_term_b, const float *direct, float *r_hprev, size_t rows, size_t hidden, float eps) override	pulsatrix::CUDABackend	virtual
im2col(const float *in, float *col, size_t n, size_t c, size_t h, size_t w, const ConvGeometry &geometry) override	pulsatrix::CUDABackend	virtual
layer_norm_backward(const float *grad_out, const float *gamma, const float *xhat, const float *row_std, float *grad_in, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
layer_norm_forward(const float *in, const float *gamma, const float *beta, float *xhat, float *out, float *row_std, size_t rows, size_t cols, float eps) override	pulsatrix::CUDABackend	virtual
logic_pointwise(LogicOp op, int norm, const float *a, const float *b, const float *g_or_r, const float *y, float *out_a, float *out_b, size_t n, float eps) override	pulsatrix::CUDABackend	virtual
logsumexp_rows(const float *in, float *out, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
lrp_avg_pool(const float *x, const float *r, float *r_in, size_t planes, size_t h, size_t w, size_t kh, size_t kw, float eps) override	pulsatrix::CUDABackend	virtual
lrp_bilinear_elementwise(const float *a, const float *b, const float *r, float *r_out, size_t n, float eps) override	pulsatrix::CUDABackend	virtual
lrp_bilinear_matmul(const float *a, const float *b, const float *o, const float *r_o, float *r_a, float *r_b, size_t slices, size_t m, size_t p, size_t q, float eps, bool b_transposed) override	pulsatrix::CUDABackend	virtual
lrp_conv(const float *col, const float *kernel, const float *pre_bias, const float *r, float *r_col, size_t n, size_t out_channels, size_t p, size_t q, float eps) override	pulsatrix::CUDABackend	virtual
lrp_linear(const float *x, const float *w, const float *z, const float *r, float *r_in, size_t rows, size_t in_features, size_t out_features, float eps) override	pulsatrix::CUDABackend	virtual
lrp_residual_split(const float *a, const float *b, const float *r, float *r_a, float *r_b, size_t n, float eps) override	pulsatrix::CUDABackend	virtual
lrp_rope(const float *x, const float *y, const float *r, const float *cos_table, const float *sin_table, float *r_in, size_t slices, size_t seq_len, size_t head_dim, float eps) override	pulsatrix::CUDABackend	virtual
lrp_softmax_rows(const float *x, const float *y, const float *r, float *r_in, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
lrp_stabilized_divide(const float *r, const float *denom, const float *gate, float *out, size_t n, float eps, LrpGate gate_mode) override	pulsatrix::CUDABackend	virtual
max_pool_forward(const float *in, float *out, float *argmax, size_t planes, size_t h, size_t w, size_t kh, size_t kw) override	pulsatrix::CUDABackend	virtual
max_unpool(const float *src, const float *argmax, float *dst, size_t planes, size_t h, size_t w, size_t kh, size_t kw) override	pulsatrix::CUDABackend	virtual
mul(const float *a, const float *b, float *out, size_t n) override	pulsatrix::CUDABackend	virtual
operator=(const CUDABackend &)=delete	pulsatrix::CUDABackend
permute_0213(const float *in, float *out, size_t d0, size_t d1, size_t d2, size_t d3) override	pulsatrix::CUDABackend	virtual
recurrent_cell(RecurrentCellOp op, const RecurrentCellArgs &args, size_t n) override	pulsatrix::CUDABackend	virtual
rl_rows(RlRowOp op, const RlRowArgs &args) override	pulsatrix::CUDABackend	virtual
rms_norm_backward(const float *grad_out, const float *gamma, const float *in, const float *row_rms, float *grad_in, float *gamma_terms, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
rms_norm_forward(const float *in, const float *gamma, float *out, float *row_rms, size_t rows, size_t cols, float eps) override	pulsatrix::CUDABackend	virtual
rope_rotate(const float *in, const float *cos_table, const float *sin_table, float *out, size_t num_slices, size_t seq_len, size_t head_dim, bool inverse) override	pulsatrix::CUDABackend	virtual
scatter_add_rows(const float *src, const float *indices, float *table, size_t count, size_t dim) override	pulsatrix::CUDABackend	virtual
softmax_rows(const float *in, float *out, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
softmax_rows_backward(const float *y, const float *dy, float *dx, size_t rows, size_t cols) override	pulsatrix::CUDABackend	virtual
ssm_pass(SsmPassOp op, const SsmPassArgs &args) override	pulsatrix::CUDABackend	virtual
sum(const float *in, size_t n) override	pulsatrix::CUDABackend	virtual
tanh_gaussian_backward(const float *action, const float *std_cache, const float *eps, const float *grad_action, const float *grad_log_prob, float *grad_mean, float *grad_log_std, size_t n, float stabilizer) override	pulsatrix::CUDABackend	virtual
tanh_gaussian_forward(const float *mean, const float *log_std, const float *eps, float *action, float *std_cache, float *log_prob, size_t rows, size_t cols, float stabilizer, double half_log_two_pi) override	pulsatrix::CUDABackend	virtual
top_k_rows(const float *in, float *values, float *indices, size_t rows, size_t cols, size_t k, bool largest) override	pulsatrix::CUDABackend	virtual
~CUDABackend() override	pulsatrix::CUDABackend
~DeviceBackend()=default	pulsatrix::DeviceBackend	virtual