From 6055bb3163414f84c818926e1573a7cc17669bc5 Mon Sep 17 00:00:00 2001 From: Ricardo Wurmus Date: Wed, 27 Dec 2023 23:27:12 +0100 Subject: 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(+) (limited to 'gnu/packages/machine-learning.scm') 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") -- cgit v1.2.3