文件名称:onnx-plaidml:使用PlaidML的ONNX后端
文件大小:34KB
文件格式:ZIP
更新时间:2024-06-03 09:15:41
Python
ONNX格子ML 是用于深度神经网络的高效计算引擎; 是神经网络的一种交换格式,并提供了一个直接API,可与各种神经网络框架集成。 该程序包在两者之间架起了桥梁。 它包括: 一个PlaidML ONNX后端,允许将ONNX用作高级后端的每个神经网络框架都将PlaidML用作其低级计算层。 一个ONNX-to-PlaidML文件转换器,使应用程序可以轻松地直接通过PlaidML API运行模型,从而提高了性能并减少了依赖性(例如Python)。 当前状态 从此版本开始,我们已经实现了我们关注的网络所需的操作,定义相对简单明了的操作以及ONNX后端测试执行的大多数操作。 ONNX后端测试并没有很好地涵盖许多操作,我们还没有为它们实现自己的测试(我们认为将它们添加到ONNX会更有用); 可能有错误。 我们认为我们已经实现了足够多的操作才有用,但是我们知道我们缺少了一些-特别是卷积的边
【文件预览】:
onnx-plaidml-master
----setup.py(118B)
----.gitignore(74B)
----onnx_plaidml()
--------__init__.py(2KB)
--------backend.py(11KB)
--------opset_util.py(5KB)
--------conversion.py(661B)
--------opset_onnx.py(44KB)
----requirements.txt(21B)
----.intel-ocl-silent(219B)
----.travis.yml(733B)
----LICENSE(34KB)
----setup.cfg(1KB)
----README.md(5KB)
----tests()
--------__init__.py(0B)
--------test_onnx_backend.py(4KB)
----.appveyor.yml(741B)
----.travis()
--------before_install.sh(1KB)