-
Notifications
You must be signed in to change notification settings - Fork 335
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
Custom Op runtime wrapper example #150
base: main
Are you sure you want to change the base?
Conversation
#include "custom_op.h" | ||
#include "openvino_wrapper.h" | ||
|
||
static const char* c_OpDomain = "ai.onnx.contrib"; |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
TODO: remake onnx model to use a different domain name
c_cxx/OpWrapper_EP/OpenVINO/main.cpp
Outdated
|
||
[[nodiscard]] bool HasArgs() const { return index_ < argc_; } | ||
|
||
[[nodiscard]] std::string_view NextArg() { |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Change to Next()
c_cxx/OpWrapper_EP/OpenVINO/main.cpp
Outdated
|
||
// Setup input | ||
std::array<int64_t, 4> input_shape{1, 3, 224, 224}; | ||
std::vector<float> input_vals(GetShapeSize(input_shape), 1.0f); |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
TODO: Input image data.
This is a draft to illustrate the use of the tentatively named OpWrapperExecutionProvider: microsoft/onnxruntime#13216
This runs an OpenVINO classification example using a custom OP that wraps a googlenet-v1 model.
TODO