From 6055bb3163414f84c818926e1573a7cc17669bc5 Mon Sep 17 00:00:00 2001
From: Ricardo Wurmus <rekado@elephly.net>
Date: Wed, 27 Dec 2023 23:27:12 +0100
Subject: [PATCH] gnu: Add onnx-optimizer-for-torch2.

* gnu/packages/machine-learning.scm (onnx-optimizer-for-torch2): New variable.

Change-Id: Icf48cba3f0a30f171b6f01cd033dd135a078ba1a
---
 gnu/packages/machine-learning.scm | 7 +++++++
 1 file changed, 7 insertions(+)

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")