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

[WebNN EP] Skip split initializer (#18729)

Browse source
This commit is contained in:
Wanming Lin 2023-12-08 00:09:49 +08:00 committed by GitHub
parent 49470f06e8
commit 3d8af6eb65
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 12 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

12
onnxruntime/core/providers/webnn/builders/impl/split_op_builder.cc
View file

@ -17,6 +17,9 @@ namespace webnn {
class SplitOpBuilder : public BaseOpBuilder {
// Add operator related.
public:
void AddInitializersToSkip(ModelBuilder& model_builder, const Node& node) const override;
private:
Status AddToModelBuilderImpl(ModelBuilder& model_builder, const Node& node,
const logging::Logger& logger) const override ORT_MUST_USE_RESULT;
@ -29,6 +32,15 @@ class SplitOpBuilder : public BaseOpBuilder {
int GetMinSupportedOpSet(const Node& node) const override;
};
// Add operator related.
void SplitOpBuilder::AddInitializersToSkip(ModelBuilder& model_builder, const Node& node) const {
// Skip split initializer if present.
if (node.InputDefs().size() > 1) {
model_builder.AddInitializerToSkip(node.InputDefs()[1]->Name());
}
}
Status SplitOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder,
const Node& node,
const logging::Logger& logger) const {