diff --git a/benchmark/opperf/utils/op_registry_utils.py b/benchmark/opperf/utils/op_registry_utils.py index 66d08cea6df3..9b0beb6eab20 100644 --- a/benchmark/opperf/utils/op_registry_utils.py +++ b/benchmark/opperf/utils/op_registry_utils.py @@ -365,7 +365,7 @@ def get_all_loss_operators(): # Filter for NN Loss operators loss_mx_operators = {} for op_name, op_params in mx_operators.items(): - if op_name in loss_ops and op_name not in unique_ops: + if op_name in loss_ops: loss_mx_operators[op_name] = mx_operators[op_name] return loss_mx_operators