diff options
| author | Ricardo Wurmus <rekado@elephly.net> | 2023-12-27 23:27:12 +0100 |
|---|---|---|
| committer | Ricardo Wurmus <rekado@elephly.net> | 2024-01-01 14:35:12 +0100 |
| commit | 6055bb3163414f84c818926e1573a7cc17669bc5 (patch) | |
| tree | ebae7571f00b50b837ea46c459f341d05d08b498 /gnu/packages/machine-learning.scm | |
| parent | 43c92f323e1750d2d3661b419e53c45c673501af (diff) | |
gnu: Add onnx-optimizer-for-torch2.
* gnu/packages/machine-learning.scm (onnx-optimizer-for-torch2): New variable.
Change-Id: Icf48cba3f0a30f171b6f01cd033dd135a078ba1a
Diffstat (limited to 'gnu/packages/machine-learning.scm')
| -rw-r--r-- | gnu/packages/machine-learning.scm | 7 |
1 files changed, 7 insertions, 0 deletions
diff --git a/gnu/packages/machine-learning.scm b/gnu/packages/machine-learning.scm index bef4d62163..6455fcc024 100644 --- a/gnu/packages/machine-learning.scm +++ b/gnu/packages/machine-learning.scm @@ -1282,6 +1282,13 @@ aim is to provide all such passes along with ONNX so that they can be re-used with a single function call.") (license license:expat))) +(define-public onnx-optimizer-for-torch2 + (package + (inherit onnx-optimizer) + (inputs + (modify-inputs (package-inputs onnx-optimizer) + (replace "onnx" onnx-for-torch2))))) + (define-public rxcpp (package (name "rxcpp") |