From 98c7d0788c6bdf68cbaa16cc70807cec2afbf7e5 Mon Sep 17 00:00:00 2001 From: Connor Goggins Date: Wed, 29 Jan 2020 16:40:46 -0800 Subject: [PATCH] Moved ElementWiseSum to binary_element_wise_operators --- benchmark/opperf/utils/op_registry_utils.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/benchmark/opperf/utils/op_registry_utils.py b/benchmark/opperf/utils/op_registry_utils.py index 4e3e7eb1264f..43158dbc4edf 100644 --- 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