Skip to main content

Module log_elementwise

Module log_elementwise 

Source
Expand description

Elementwise natural log forward + backward.

Used by reverse-mode autograd in downstream crates (hf2q ADR-020 Track 1: log_softmax + KL-div composition).

Statics§

LOG_SHADER_SOURCE

Functions§

dispatch_log_backward_f32
Encode dx[i] = dy[i] / x[i] (the backward pass for elementwise log). x is the FORWARD INPUT, not the forward output.
dispatch_log_f32
Encode output[i] = log(input[i]) for f32 input.
register