saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-09 17:28:58 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

Fix CPU Xor implementation (#8934)

Browse source
This commit is contained in:
Hariharan Seshadri 2021-09-01 21:38:55 -07:00 committed by GitHub
parent 1985616262
commit ddbc8bc5fc
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
2 changed files with 10 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/cpu/math/element_wise_ops.cc
View file

@ -898,7 +898,7 @@ Status Xor::Compute(OpKernelContext* context) const {
ProcessBroadcastSpanFuncs funcs{
[](BroadcastHelper& per_iter_bh) {
bool input0 = per_iter_bh.ScalarInput0<bool>();
auto input1 = per_iter_bh.EigenInput0<bool>();
auto input1 = per_iter_bh.EigenInput1<bool>();
auto output = per_iter_bh.OutputEigen<bool>();
if (input0)
output.array() = !input1.array();

9
onnxruntime/test/providers/cpu/math/element_wise_ops_test.cc
View file

@ -1553,6 +1553,15 @@ TEST(MathOpTest, Xor) {
test.Run();
}
TEST(MathOpTest, Xor_Issue8880) {
OpTester test("Xor");
test.AddInput<bool>("A", {1}, {true});
test.AddInput<bool>("B", {1}, {false});
test.AddOutput<bool>("C", {1}, {true});
test.Run();
}
TEST(MathOpTest, Xor_bcast3v2d) {
OpTester test("Xor");