-
Notifications
You must be signed in to change notification settings - Fork 2.4k
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
[ONNX FE] Add SoftmaxCrossEntropyLoss operator to ONNX Frontend #28682
base: master
Are you sure you want to change the base?
[ONNX FE] Add SoftmaxCrossEntropyLoss operator to ONNX Frontend #28682
Conversation
Hi @gkrivor |
|
||
const auto scores = inputs[0]; | ||
const auto labels = inputs[1]; | ||
|
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.
I think weights
are also needed according to the spec. Please check.
Hello! It does have a CMakeLists.txt file, so I'm wondering if I could save time by doing that. |
Please bear with me for a while. My laptop's number keys just gave up and do not function anymore. And I have no external keyboard at the moment. It's going to take a while for my next update, after the servicing is over. |
This pull request implements the SoftmaxCrossEntropyLoss operation for the ONNX frontend in OpenVINO, supporting opset 12 and opset 13. It includes new header and source files to define and implement the operation.
Ticket #20547
New operation implementation:
src/frontends/onnx/frontend/src/op/softmax_cross_entropy_loss.hpp
: Added the header file defining theSoftmaxCrossEntropyLoss
operation for opset 12 and opset 13.src/frontends/onnx/frontend/src/op/softmax_crossentropy_loss.cpp
: Added the source file implementing theSoftmaxCrossEntropyLoss
operation, including the shared helper functionimpl_softmax_cross_entropy
to compute the loss.TODO
softmax_cross_entropy_loss.cpp
&softmax_cross_entropy_loss.hpp
opset_12
&opset_13
ONNX_OP()
macroP.S.: I'm very new to contributing to large codebases, and I want to know if my work meets the standards and most of all, is what's expected!