diff options
| -rw-r--r-- | var/spack/repos/builtin/packages/py-keras2onnx/package.py | 24 | ||||
| -rw-r--r-- | var/spack/repos/builtin/packages/py-onnxconverter-common/package.py | 20 |
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')) |