diff --git a/onnxruntime/core/providers/webnn/builders/helper.h b/onnxruntime/core/providers/webnn/builders/helper.h index 68f009a94e..73e3008621 100644 --- a/onnxruntime/core/providers/webnn/builders/helper.h +++ b/onnxruntime/core/providers/webnn/builders/helper.h @@ -212,7 +212,7 @@ static const InlinedHashMap op_map = { {"Tanh", {"tanh", true}}, {"Transpose", {"transpose", true}}, {"Unsqueeze", {"reshape", true}}, - {"Where", {"elementwiseIf", false}}, + {"Where", {"where", false}}, }; inline bool CheckSingleOp(const std::string& op_type, const emscripten::val& wnn_builder_, diff --git a/onnxruntime/core/providers/webnn/builders/impl/ternary_op_builder.cc b/onnxruntime/core/providers/webnn/builders/impl/ternary_op_builder.cc index e51c17fc56..9c23554a44 100644 --- a/onnxruntime/core/providers/webnn/builders/impl/ternary_op_builder.cc +++ b/onnxruntime/core/providers/webnn/builders/impl/ternary_op_builder.cc @@ -32,7 +32,7 @@ Status TernaryOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder, cons emscripten::val input2 = model_builder.GetOperand(node.InputDefs()[2]->Name()); emscripten::val output = emscripten::val::object(); if (op_type == "Where") { - output = model_builder.GetBuilder().call("elementwiseIf", input0, input1, input2); + output = model_builder.GetBuilder().call("where", input0, input1, input2); } else { return ORT_MAKE_STATUS(ONNXRUNTIME, INVALID_ARGUMENT, "TernaryOpBuilder::AddToModelBuilderImpl, unknown op: ", op_type);