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

[WebNN EP] Support PRelu op (#16756)

Browse source
This commit is contained in:
Wanming Lin 2023-07-21 01:39:30 +08:00 committed by GitHub
parent bb136f86c8
commit eaea34f8e2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
3 changed files with 5 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

1
onnxruntime/core/providers/webnn/builders/helper.h
View file

@ -167,6 +167,7 @@ static const InlinedHashMap<std::string, std::string> op_map = {
{"Not", "logicalNot"},
{"Pad", "pad"},
{"Pow", "pow"},
{"PRelu", "prelu"},
{"Reciprocal", "reciprocal"},
{"ReduceMax", "reduceMax"},
{"ReduceMean", "reduceMean"},

3
onnxruntime/core/providers/webnn/builders/impl/binary_op_builder.cc
View file

@ -39,6 +39,8 @@ Status BinaryOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder, const
output = model_builder.GetBuilder().call<emscripten::val>("div", input0, input1);
} else if (op_type == "Pow") {
output = model_builder.GetBuilder().call<emscripten::val>("pow", input0, input1);
} else if (op_type == "PRelu") {
output = model_builder.GetBuilder().call<emscripten::val>("prelu", input0, input1);
} else {
return ORT_MAKE_STATUS(ONNXRUNTIME, INVALID_ARGUMENT,
"BinaryOpBuilder::AddToModelBuilderImpl, unknown op: ", op_type);
@ -59,6 +61,7 @@ void CreateBinaryOpBuilder(const std::string& op_type, OpBuilderRegistrations& o
"Mul",
"Div",
"Pow",
"PRelu",
};
op_registrations.builders.push_back(std::make_unique<BinaryOpBuilder>());

1
onnxruntime/core/providers/webnn/builders/op_builder_factory.cc
View file

@ -37,6 +37,7 @@ static OpBuilderRegistrations CreateOpBuilderRegistrations() {
CreateBinaryOpBuilder("Mul", op_registrations);
CreateBinaryOpBuilder("Div", op_registrations);
CreateBinaryOpBuilder("Pow", op_registrations);
CreateBinaryOpBuilder("PRelu", op_registrations);
}
{ // Ternary