File tree Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Expand file tree Collapse file tree 1 file changed +3
-5
lines changed Original file line number Diff line number Diff line change @@ -66,12 +66,10 @@ You can also pass a numpy array here.
66
66
67
67
### Inline some weights in generated stablehlo
68
68
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.
72
70
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.
75
73
76
74
So as above, the function we exported ` jfunc ` takes ` weights ` and ` args ` as input, so
77
75
they appear as paramters.
You can’t perform that action at this time.
0 commit comments