aboutsummaryrefslogtreecommitdiff
diff options
context:
space:
mode:
-rw-r--r--gnu/packages/machine-learning.scm7
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")