Moved ElementWiseSum to binary_element_wise_operators

apache · Feb 5, 2020 · 98c7d07 · 98c7d07
1 parent e76dd3d
commit 98c7d07
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/benchmark/opperf/utils/op_registry_utils.py b/benchmark/opperf/utils/op_registry_utils.py
@@ -222,8 +222,6 @@ def get_all_misc_binary_operators():
             binary_misc_mx_operators[op_name] = mx_operators[op_name]
         elif "reshape_like" == op_name:
             binary_misc_mx_operators[op_name] = mx_operators[op_name]
-        elif "ElementWiseSum" == op_name:
-            binary_misc_mx_operators[op_name] = mx_operators[op_name]
     return binary_misc_mx_operators
 
 
@@ -244,6 +242,8 @@ def get_all_elemen_wise_binary_operators():
                 "lhs" in op_params["params"]["arg_names"] and \
                 "rhs" in op_params["params"]["arg_names"]:
             binary_elemen_wise_mx_operators[op_name] = mx_operators[op_name]
+        elif "ElementWiseSum" == op_name:
+            binary_elemen_wise_mx_operators[op_name] = mx_operators[op_name]
     return binary_elemen_wise_mx_operators