diff --git a/benchmark/opperf/utils/op_registry_utils.py b/benchmark/opperf/utils/op_registry_utils.py index 717f495bd748..a2149d665c19 100644 --- a/benchmark/opperf/utils/op_registry_utils.py +++ b/benchmark/opperf/utils/op_registry_utils.py @@ -121,8 +121,6 @@ def prepare_op_inputs(op, arg_params): # 3d tensor is needed by following ops ops_3d = ['CTCLoss', 'ctc_loss'] - # For args that need to change shape/value for different ops - custom_args = ['act_type'] # For ops with args that need to change shape/value for different ops custom_data = ['Activation', 'LeakyReLU', 'Softmax'] @@ -151,8 +149,6 @@ def prepare_op_inputs(op, arg_params): # This is for cases where in some ops 'axis' is Int in some ops a shape tuple. # Ex: axis in sum is shape, axis in sort is int. arg_values[arg_name] = DEFAULTS_INPUTS[arg_name + "_shape"] - elif arg_name in custom_args and arg_name + "_" + op.lower() in DEFAULTS_INPUTS: - arg_values[arg_name] = DEFAULTS_INPUTS[arg_name + "_" + op.lower()] # Number of different inputs we want to use to test # the operator