Skip to content

Commit

Permalink
Update to ONNX Runtime 1.16.0 (#340)
Browse files Browse the repository at this point in the history
* Testing ORT 1.15.0-rc1

* Updating to 1.16-rc.

* Moving to released ort 1.16.0.
  • Loading branch information
Craigacp authored Dec 1, 2023
1 parent af8dac4 commit e6d310a
Show file tree
Hide file tree
Showing 4 changed files with 5 additions and 8 deletions.
5 changes: 0 additions & 5 deletions Interop/ONNX/pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -94,11 +94,6 @@
</dependency>
</dependencies>

<properties>
<maven.compiler.source>1.8</maven.compiler.source>
<maven.compiler.target>1.8</maven.compiler.target>
</properties>

<build>
<plugins>
<plugin>
Expand Down
Original file line number Diff line number Diff line change
Expand Up @@ -17,6 +17,7 @@
package org.tribuo.interop.onnx;

import ai.onnxruntime.OnnxJavaType;
import ai.onnxruntime.OnnxMap;
import ai.onnxruntime.OnnxSequence;
import ai.onnxruntime.OnnxTensor;
import ai.onnxruntime.OnnxValue;
Expand Down Expand Up @@ -184,8 +185,9 @@ protected float[][] getBatchPredictions(List<OnnxValue> inputs, ImmutableOutputI
float[][] outputArray = new float[output.size()][outputIDInfo.size()];
int i = 0;
for (Object o : output) {
OnnxMap onnxMap = (OnnxMap) o;
@SuppressWarnings("unchecked") // guarded by the if on the mapInfo above.
Map<Long,Float> map = (Map<Long,Float>) o;
Map<Long,Float> map = (Map<Long,Float>) onnxMap.getValue();
if (map.size() == outputIDInfo.size()) {
for (Map.Entry<Long,Float> e : map.entrySet()) {
Long key = e.getKey();
Expand Down
2 changes: 1 addition & 1 deletion THIRD_PARTY_LICENSES.txt
Original file line number Diff line number Diff line change
Expand Up @@ -89,7 +89,7 @@ LIABILITY, WHETHER IN CONTRACT, STRICT LIABILITY, OR TORT (INCLUDING
NEGLIGENCE OR OTHERWISE) ARISING IN ANY WAY OUT OF THE USE OF THIS
SOFTWARE, EVEN IF ADVISED OF THE POSSIBILITY OF SUCH DAMAGE.

onnx-runtime 1.12.1 - MIT License
onnx-runtime 1.16.0 - MIT License

MIT License

Expand Down
2 changes: 1 addition & 1 deletion pom.xml
Original file line number Diff line number Diff line change
Expand Up @@ -51,7 +51,7 @@
<!-- 3rd party backend dependencies -->
<liblinear.version>2.44</liblinear.version>
<libsvm.version>3.25</libsvm.version>
<onnxruntime.version>1.12.1</onnxruntime.version>
<onnxruntime.version>1.16.0</onnxruntime.version>
<tensorflow.version>0.5.0</tensorflow.version>
<xgboost.version>1.6.2</xgboost.version>

Expand Down

0 comments on commit e6d310a

Please sign in to comment.