-
-
Notifications
You must be signed in to change notification settings - Fork 110
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
WIP: Custom operators #156
Conversation
Codecov ReportAttention:
Additional details and impacted files@@ Coverage Diff @@
## main #156 +/- ##
==========================================
- Coverage 50.68% 44.48% -6.21%
==========================================
Files 15 19 +4
Lines 1241 1423 +182
==========================================
+ Hits 629 633 +4
- Misses 612 790 +178 ☔ View full report in Codecov by Sentry. |
…nto custom-operators
Co-authored-by: Varun Chatterji <[email protected]>
- Update existing documentation to be relevant to v2.0 changes - Add missing documentation and doctests/examples - Remove unused code - Replace most pointer usage with `NonNull` - Use slightly safer pointer `.cast()`s - Disallow extracting string tensors with `extract_raw_tensor` (TODO: owned variant of this function) - Add lifetime bound to `ModelMetadata` to fix a potential use-after-free - Add per-run tag to `RunOptions` - Replace `SessionInputKey` with `Cow<str>` - Register execution providers inside `SessionBuilder::with_execution_providers`, not on commit
- Update existing documentation to be relevant to v2.0 changes - Add missing documentation and doctests/examples - Remove unused code - Replace most pointer usage with `NonNull` - Use slightly safer pointer `.cast()`s - Disallow extracting string tensors with `extract_raw_tensor` (TODO: owned variant of this function) - Add lifetime bound to `ModelMetadata` to fix a potential use-after-free - Add per-run tag to `RunOptions` - Replace `SessionInputKey` with `Cow<str>` - Register execution providers inside `SessionBuilder::with_execution_providers`, not on commit
Allows custom operator kernels to be written and used entirely in Rust. See here for an example of the API.
TODO
GetKernelAttribute
for all supported attribute types