Struct GridSampleBackwardPlan 

pub struct GridSampleBackwardPlan<T: Element> { /* private fields */ }

grid_sample_backward plan.

Adjoint of crate::GridSamplePlan: scatter dout into both dinput (4 bilinear weights via atomicAdd) and dgrid (analytic chain rule through the coordinate mapping).

When to use: BW for GridSamplePlan. Caller retains FW input and grid.

Dtypes: {f32, f64}.

Shape limits: rank-4 NCHW + grid [N, OH, OW, 2].

Workspace: none. Caller MUST zero dinput and dgrid before launch.

Precision guarantee: non-deterministic (atomicAdd into dinput).

Implementations

impl<T: Element> GridSampleBackwardPlan<T>

pub fn select( _stream: &Stream, desc: &GridSampleBackwardDescriptor, _pref: PlanPreference, ) -> Result<Self>

Pick a kernel.

pub fn can_implement(&self, args: &GridSampleBackwardArgs<'_, T>) -> Result<()>

Validate args.

pub fn workspace_size(&self) -> usize

Workspace (zero).

pub fn sku(&self) -> KernelSku

Identity.

pub fn precision_guarantee(&self) -> PrecisionGuarantee

Numerical guarantees.

pub fn run( &self, stream: &Stream, _workspace: Workspace<'_>, args: GridSampleBackwardArgs<'_, T>, ) -> Result<()>

Launch.