gausplat_renderer::render::gaussian_3d::jit

Function backward

source 
pub fn backward<R: JitRuntime, F: FloatElement, I: IntElement, B: BoolElement>(
    state: RenderInput<JitBackend<R, F, I, B>>,
    colors_rgb_2d_grad: JitTensor<R>,
) -> RenderOutput<JitBackend<R, F, I, B>>
Expand description

Render the 3DGS scene (backward).

It computes the gradients of colors_rgb_2d with respect to scene parameters.

The kernels are launched in the following order:

  1. Compute the gradients for rasterization.
  2. Compute the gradients for transformation.