Skip to content

Commit 0f872a3

Browse files
committed
mikegre/s chagne
1 parent 9aefcf8 commit 0f872a3

File tree

1 file changed

+3
-5
lines changed

1 file changed

+3
-5
lines changed

docs/source/features/stablehlo.md

Lines changed: 3 additions & 5 deletions
Original file line numberDiff line numberDiff line change
@@ -66,12 +66,10 @@ You can also pass a numpy array here.
6666

6767
### Inline some weights in generated stablehlo
6868

69-
Suppose that you want to inline some (or all) of the model's weight
70-
into the generated StableHLO graph as constant. You can accomplish it by
71-
exporting a different function that calls your model.
69+
You can inline some or all of your model's weights into the StableHLO graph as constants by exporting a separate function that calls your model.
7270

73-
The convention used in `jax.jit` is that, all the input of the `jit`'d python
74-
functions are exported as parameters, and everything else are inlined as constants.
71+
The convention used in `jax.jit` is all the input of the `jit`ed Python
72+
functions are exported as parameters, everything else are inlined as constants.
7573

7674
So as above, the function we exported `jfunc` takes `weights` and `args` as input, so
7775
they appear as paramters.

0 commit comments

Comments
 (0)