Skip to content

Commit

Permalink
[Opperf] Filter out deprecated ops (apache#15541)
Browse files Browse the repository at this point in the history
* normal, uniform ops

* Revert "normal, uniform ops"

This reverts commit f8d6f95.

* filter out deprecated ops

* Trigger notification

* additional deprecated ops

* Trigger notification

* Trigger notification

* Trigger notification
  • Loading branch information
ChaiBapchya authored and Ubuntu committed Aug 20, 2019
1 parent daf16ae commit 4d729c8
Showing 1 changed file with 7 additions and 0 deletions.
7 changes: 7 additions & 0 deletions benchmark/opperf/utils/op_registry_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -37,6 +37,8 @@ def _select_ops(operator_names, filters=("_contrib", "_"), merge_op_forward_back
By default, filter out all Contrib operators that starts with '_contrib' and internal operators that
starts with '_'.
Note - All deprecated operators are filtered out as well.
Parameters
----------
operator_names: List[str]
Expand All @@ -53,6 +55,11 @@ def _select_ops(operator_names, filters=("_contrib", "_"), merge_op_forward_back
mx_operators = {}
operators_with_backward = []

# Filter out deprecated operators
filters += ("normal", "uniform", "BatchNorm_v1", "Flatten", "contrib_CTCLoss", "Pad", "Cast",
"Pooling_v1", "Concat", "Reshape", "Convolution_v1", "SliceChannel", "Crop",
"crop", "onehot_encode")

if merge_op_forward_backward:
filters += ("_backward",)

Expand Down

0 comments on commit 4d729c8

Please sign in to comment.