Merge pull request #23549 from pschuh:docs-update

PiperOrigin-RevId: 673148286
jax-ml · Sep 10, 2024 · 14b8625 · 14b8625
2 parents 0b04dd0 + 030b6c6
commit 14b8625
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/jax/_src/lax/convolution.py b/jax/_src/lax/convolution.py
@@ -114,7 +114,7 @@ def conv_general_dilated(
   - the input and output feature dimensions in rhs with the characters 'I'
     and 'O' respectively, and
   - spatial dimension correspondences between lhs, rhs, and the output using
-    any distinct characters.
+    any distinct characters. The examples below use 'W' and 'H'.
 
   For example, to indicate dimension numbers consistent with the ``conv``
   function with two spatial dimensions, one could use ``('NCHW', 'OIHW',

diff --git a/jax/_src/lax/other.py b/jax/_src/lax/other.py
@@ -187,7 +187,7 @@ def conv_general_dilated_local(
   - the input and output feature dimensions in rhs with the characters 'I'
     and 'O' respectively, and
   - spatial dimension correspondences between `lhs`, `rhs`, and the output using
-    any distinct characters.
+    any distinct characters. The examples below use 'W' and 'H'.
 
   For example, to indicate dimension numbers consistent with the `conv` function
   with two spatial dimensions, one could use `('NCHW', 'OIHW', 'NCHW')`. As