Fix incorrect argument name in batch_norm_in_place doc comment

robertknight · robertknight · commit 7c24b39b1fc7 · 2025-04-12T18:40:06.000+01:00
diff --git a/src/ops/norm.rs b/src/ops/norm.rs
@@ -177,7 +177,7 @@ fn normalize_each_channel<'a>(
     }
 }
 
-/// Perform in-place batch normalization on the `NC*` tensor `out`.
+/// Perform in-place batch normalization on an `NC*` tensor.
 ///
 /// See <https://github.com/onnx/onnx/blob/main/docs/Operators.md#batchnormalization>.
 pub fn batch_norm_in_place(
@@ -206,7 +206,7 @@ pub fn batch_norm_in_place(
     Ok(())
 }
 
-/// Perform batch normalization on the `NC*` tensor `input`.
+/// Perform batch normalization on an `NC*` tensor.
 ///
 /// See <https://github.com/onnx/onnx/blob/main/docs/Operators.md#batchnormalization>.
 pub fn batch_norm(

Original file line number	Diff line number	Diff line change
`@@ -177,7 +177,7 @@ fn normalize_each_channel<'a>(`
`177`	`177`	`}`
`178`	`178`	`}`
`179`	`179`
`180`		-/// Perform in-place batch normalization on the `NC*` tensor `out`.
	`180`	+/// Perform in-place batch normalization on an `NC*` tensor.
`181`	`181`	`///`
`182`	`182`	`/// See <https://github.com/onnx/onnx/blob/main/docs/Operators.md#batchnormalization>.`
`183`	`183`	`pub fn batch_norm_in_place(`
`@@ -206,7 +206,7 @@ pub fn batch_norm_in_place(`
`206`	`206`	`Ok(())`
`207`	`207`	`}`
`208`	`208`
`209`		-/// Perform batch normalization on the `NC*` tensor `input`.
	`209`	+/// Perform batch normalization on an `NC*` tensor.
`210`	`210`	`///`
`211`	`211`	`/// See <https://github.com/onnx/onnx/blob/main/docs/Operators.md#batchnormalization>.`
`212`	`212`	`pub fn batch_norm(`