Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Moving ONNX export utils out into a separate module #203

Merged
merged 1 commit into from
Dec 13, 2021
Merged

Conversation

Craigacp
Copy link
Member

Description

Moves the org.tribuo.onnx classes from tribuo-core to tribuo-util-onnx, which has no dependencies on the rest of Tribuo. Then moved ONNXExportable to org.tribuo, and moved the model proto construction method to ONNXExportable from ONNXContext.

Motivation

Splitting out the core ONNX model writing code into a separate jar will make it possible for the rest of the JVM ecosystem to use it to build out ONNX model support without inducing a Tribuo dependency. This will make it easier if we decide to upstream that support later, and will hopefully encourage contributions to fill out the support with the rest of the ONNX operators.

@Craigacp Craigacp added Oracle employee This PR is from an Oracle employee squash-commits Squash the commits when merging this PR labels Dec 10, 2021
Copy link
Member

@JackSullivan JackSullivan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

A minor question about ONNXExportable interface, otherwise looks good.

Copy link
Member

@JackSullivan JackSullivan left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

Looks good to me.

@Craigacp Craigacp merged commit a6f48f8 into main Dec 13, 2021
@Craigacp Craigacp deleted the onnx-module branch December 13, 2021 16:12
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
Oracle employee This PR is from an Oracle employee squash-commits Squash the commits when merging this PR
Projects
None yet
Development

Successfully merging this pull request may close these issues.

2 participants