summaryrefslogtreecommitdiff
diff options
context:
space:
mode:
authoriarspider <iarspider@gmail.com>2021-12-13 20:47:27 +0100
committerGitHub <noreply@github.com>2021-12-13 12:47:27 -0700
commitc1d51593d1d0022f253ac3c5b5a2c13009bf1bd6 (patch)
tree51bbcb0f3edb2e1dc1ae3adb08e9be258833187b
parent948bc98fa6c2cba0894fdaf5f1cd6739c63715c3 (diff)
downloadspack-c1d51593d1d0022f253ac3c5b5a2c13009bf1bd6.tar.gz
spack-c1d51593d1d0022f253ac3c5b5a2c13009bf1bd6.tar.bz2
spack-c1d51593d1d0022f253ac3c5b5a2c13009bf1bd6.tar.xz
spack-c1d51593d1d0022f253ac3c5b5a2c13009bf1bd6.zip
New packages: py-keras2onnx, py-onnxconverter-common (#27962)
* New packages: py-keras2onnx, py-onnxconverter-common * Update package.py (#18) * Update package.py * Update package.py
-rw-r--r--var/spack/repos/builtin/packages/py-keras2onnx/package.py24
-rw-r--r--var/spack/repos/builtin/packages/py-onnxconverter-common/package.py20
2 files changed, 44 insertions, 0 deletions
diff --git a/var/spack/repos/builtin/packages/py-keras2onnx/package.py b/var/spack/repos/builtin/packages/py-keras2onnx/package.py
new file mode 100644
index 0000000000..60db5d7484
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-keras2onnx/package.py
@@ -0,0 +1,24 @@
+# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack import *
+
+
+class PyKeras2onnx(PythonPackage):
+ """Converts Machine Learning models to ONNX for use in Windows ML"""
+
+ homepage = "https://github.com/onnx/keras-onnx"
+
+ url = "https://github.com/onnx/keras-onnx/archive/refs/tags/v1.7.0.tar.gz"
+
+ version('1.7.0', sha256='8ec9c4e1c1f870d420934d1aa7cbc9faab80c6af366900bf95e5f48280c0d199')
+
+ depends_on('py-setuptools', type='build')
+ depends_on('py-numpy', type=('build', 'run'))
+ depends_on('py-protobuf', type=('build', 'run'))
+ depends_on('py-requests', type=('build', 'run'))
+ depends_on('py-onnx', type=('build', 'run'))
+ depends_on('py-onnxconverter-common@1.7.0:', type=('build', 'run'))
+ depends_on('py-fire', type=('build', 'run'))
diff --git a/var/spack/repos/builtin/packages/py-onnxconverter-common/package.py b/var/spack/repos/builtin/packages/py-onnxconverter-common/package.py
new file mode 100644
index 0000000000..3024d09822
--- /dev/null
+++ b/var/spack/repos/builtin/packages/py-onnxconverter-common/package.py
@@ -0,0 +1,20 @@
+# Copyright 2013-2021 Lawrence Livermore National Security, LLC and other
+# Spack Project Developers. See the top-level COPYRIGHT file for details.
+#
+# SPDX-License-Identifier: (Apache-2.0 OR MIT)
+
+from spack import *
+
+
+class PyOnnxconverterCommon(PythonPackage):
+ """ONNX Converter and Optimization Tools"""
+
+ homepage = "https://github.com/microsoft/onnxconverter-common"
+ url = "https://github.com/microsoft/onnxconverter-common/archive/refs/tags/v1.9.0.tar.gz"
+
+ version('1.9.0', sha256='e3b0c44298fc1c149afbf4c8996fb92427ae41e4649b934ca495991b7852b855')
+
+ depends_on('py-setuptools', type='build')
+ depends_on('py-numpy', type=('build', 'run'))
+ depends_on('py-protobuf', type=('build', 'run'))
+ depends_on('py-onnx', type=('build', 'run'))