saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-07-30 20:18:08 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[CoreML EP] Fix condition in PRelu op supported check. (#11543)

Browse source
This commit is contained in:
Edward Chen 2022-05-17 09:03:24 -07:00 committed by GitHub
parent deef214772
commit 782f9e394d
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
onnxruntime/core/providers/coreml/builders/impl/activation_op_builder.cc
View file

@ -123,7 +123,7 @@ bool IsPReluOpSupported(const Node& node, const OpBuilderInputParams& input_para
}
const auto x_rank = x_shape.size();
if (x_rank == 3 || x_rank == 4) {
if (x_rank != 3 && x_rank != 4) {
LOGS(logger, VERBOSE) << "PRelu 'X' input must have 3 or 4 dimensions, it has " << x_rank << " dimensions";
return false;
}