Skip to main content

Module combined_backward

Module combined_backward 

Source

Structs§

CombinedGrads

Functions§

combined_backward
Core gradient computation. All arguments use the shapes from the forward.
k3_ssd_chunk_state_extended
Same as k3_ssd_chunk_state but return some intermediaries that are useful to the custom backward operation.