saymrwulf/onnxruntime
1
0
Fork 0
mirror of https://github.com/saymrwulf/onnxruntime.git synced 2026-05-29 23:06:41 +00:00
Code Issues Projects Releases Packages Wiki Activity Actions

[WebNN Ep] Slice's axes and steps inputs should be constant initializers (#18427)

Browse source
This commit is contained in:
Wanming Lin 2023-11-18 00:01:40 +08:00 committed by GitHub
parent 1a29460919
commit a5537f2f56
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 16 additions and 12 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

28
onnxruntime/core/providers/webnn/builders/impl/slice_op_builder.cc
View file

@ -114,6 +114,22 @@ bool SliceOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& initializers,
if (!GetShape(*input_defs[0], input_shape, logger)) {
return false;
}
if (input_defs.size() < 3) {
LOGS(logger, VERBOSE) << op_type << " [" << name << "] requires at least 3 inputs (data, starts, ends) but got "
<< input_defs.size();
return false;
}
// Inputs: starts, ends, axes, and steps must be constant initializers if present.
for (size_t i = 1; i < input_defs.size(); i++) {
if (!Contains(initializers, input_defs[i]->Name())) {
LOGS(logger, VERBOSE) << "Input [" << input_defs[i]->Name() << "] of " << op_type
<< " [" << name << "] must be known as initializer";
return false;
}
}
if (input_defs.size() == 5) { // Check steps.
const auto& steps_tensor = *initializers.at(input_defs[4]->Name());
std::vector<uint8_t> unpacked_tensor;
@ -140,18 +156,6 @@ bool SliceOpBuilder::IsOpSupportedImpl(const InitializedTensorSet& initializers,
}
}
if (input_defs.size() < 3) {
LOGS(logger, VERBOSE) << op_type << " [" << name << "] requires at least 3 inputs (data starts and ends) but got "
<< input_defs.size();
return false;
}
const auto& starts_name = input_defs[1]->Name();
const auto& ends_name = input_defs[2]->Name();
if (!Contains(initializers, starts_name) || !Contains(initializers, ends_name)) {
LOGS(logger, VERBOSE) << op_type << " [" << name << "] need starts and ends as initializer.";
return false;
}
return true;
}