Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
Dropped unnecessary custom_args
Browse files Browse the repository at this point in the history
  • Loading branch information
connorgoggins committed Feb 5, 2020
1 parent c015a59 commit f11ba22
Showing 1 changed file with 0 additions and 4 deletions.
4 changes: 0 additions & 4 deletions benchmark/opperf/utils/op_registry_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -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']
Expand Down Expand Up @@ -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
Expand Down

0 comments on commit f11ba22

Please sign in to comment.