oa::ml::matrix::channel_norm_relu_backward

Compute complete adjoints of fused channel normalization and ReLU. Errors Returns an error when saved output or gradient does not match the forward contract, or runtime recording fails.

FnRust public API

Function Documentation

pub fn channel_norm_relu_backward(
	input: &Matrix,
	weight: &Matrix,
	forward_output: &Matrix,
	output_gradient: &Matrix,
	epsilon: f32
) -> Result<ChannelNormBackward>

Compute complete adjoints of fused channel normalization and ReLU. Errors Returns an error when saved output or gradient does not match the forward contract, or runtime recording fails.