[WebNN EP] Support Greater and Less ops (microsoft#16782)

adityagoel4512 · Jul 24, 2023 · 5d17bcd · 5d17bcd
1 parent 8ede2f1
commit 5d17bcd
Show file tree

Hide file tree

Showing 3 changed files with 10 additions and 0 deletions.
diff --git a/onnxruntime/core/providers/webnn/builders/helper.h b/onnxruntime/core/providers/webnn/builders/helper.h
@@ -153,13 +153,15 @@ static const InlinedHashMap<std::string, std::string> op_map = {
     {"Gemm", "gemm"},
     {"GlobalAveragePool", "averagePool2d"},
     {"GlobalMaxPool", "maxPool2d"},
+    {"Greater", "greater"},
     {"GroupNormalization", "meanVarianceNormalization"},
     {"HardSigmoid", "hardSigmoid"},
     {"HardSwish", "hardSwish"},
     {"Identity", "identity"},
     {"InstanceNormalization", "meanVarianceNormalization"},
     {"LayerNormalization", "meanVarianceNormalization"},
     {"LeakyRelu", "leakyRelu"},
+    {"Less", "lesser"},
     {"MatMul", "matmul"},
     {"MaxPool", "maxPool2d"},
     {"Mul", "mul"},

diff --git a/onnxruntime/core/providers/webnn/builders/impl/logical_op_builder.cc b/onnxruntime/core/providers/webnn/builders/impl/logical_op_builder.cc
@@ -33,6 +33,10 @@ Status LogicalOpBuilder::AddToModelBuilderImpl(ModelBuilder& model_builder, cons
   emscripten::val output = emscripten::val::object();
   if (op_type == "Equal") {
     output = model_builder.GetBuilder().call<emscripten::val>("equal", input0, input1);
+  } else if (op_type == "Greater") {
+    output = model_builder.GetBuilder().call<emscripten::val>("greater", input0, input1);
+  } else if (op_type == "Less") {
+    output = model_builder.GetBuilder().call<emscripten::val>("lesser", input0, input1);
   } else {
     return ORT_MAKE_STATUS(ONNXRUNTIME, INVALID_ARGUMENT,
                            "LogicalOpBuilder::AddToModelBuilderImpl, unknown op: ", op_type);
@@ -49,6 +53,8 @@ void CreateLogicalOpBuilder(const std::string& op_type, OpBuilderRegistrations&
   static std::vector<std::string> op_types =
       {
           "Equal",
+          "Greater",
+          "Less",
       };
 
   op_registrations.builders.push_back(std::make_unique<LogicalOpBuilder>());

diff --git a/onnxruntime/core/providers/webnn/builders/op_builder_factory.cc b/onnxruntime/core/providers/webnn/builders/op_builder_factory.cc
@@ -97,6 +97,8 @@ static OpBuilderRegistrations CreateOpBuilderRegistrations() {
 
   {  // Logical
     CreateLogicalOpBuilder("Equal", op_registrations);
+    CreateLogicalOpBuilder("Greater", op_registrations);
+    CreateLogicalOpBuilder("Less", op_registrations);
   }
 
   {  // Normalization