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 7, 2020
1 parent d164eca commit f90099d
Showing 1 changed file with 1 addition and 5 deletions.
6 changes: 1 addition & 5 deletions benchmark/opperf/utils/op_registry_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -117,11 +117,9 @@ 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', 'BilinearSampler', 'GridGenerator', 'sample_multinomial']

# For ops with args that need to change shape/value for different ops
custom_data = ['Activation', 'LeakyReLU', 'Softmax']
custom_data = ['Activation', 'LeakyReLU', 'Softmax', 'BilinearSampler', 'GridGenerator', 'sample_multinomial']

# Prepare op to default input mapping
arg_values = {}
Expand All @@ -147,8 +145,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 f90099d

Please sign in to comment.