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

[WebNN EP] Disable clamp fusion for WebNN GPU (#18386)

Browse source 
As which has not been supported in WebNN DirectML backend.
This commit is contained in:
Wanming Lin 2023-11-13 00:56:39 +08:00 committed by GitHub
parent 8d298f6f78
commit cbf0cf06db
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 9 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

15
onnxruntime/core/providers/webnn/builders/model_builder.cc
View file

@ -64,12 +64,15 @@ void ModelBuilder::PreprocessActivations() {
const auto& op_type(node->OpType());
if (op_type == "Clip") {
float minValue, maxValue;
GetClipMinMax(GetInitializerTensors(), *node, minValue, maxValue, logger_);
emscripten::val options = emscripten::val::object();
options.set("minValue", minValue);
options.set("maxValue", maxValue);
activation_nodes_.emplace(node->Index(), wnn_builder_.call<emscripten::val>("clamp", options));
// Temporarily disable clamp fusion for WebNN GPU as which is not supported yet.
if (wnn_device_type_ == WebnnDeviceType::CPU) {
float minValue, maxValue;
GetClipMinMax(GetInitializerTensors(), *node, minValue, maxValue, logger_);
emscripten::val options = emscripten::val::object();
options.set("minValue", minValue);
options.set("maxValue", maxValue);
activation_nodes_.emplace(node->Index(), wnn_builder_.call<emscripten::val>("clamp", options));
}
} else if (op_type == "Elu") {
NodeAttrHelper helper(*node);
emscripten::val options = emscripten::val::object();