Skip to content

Commit

Permalink
Remove the deprecated BatchNorm_v1 op (apache#18538)
Browse files Browse the repository at this point in the history
* remove batchnorm_v1

* fix gpu build

Co-authored-by: EC2 Default User <[email protected]>
Co-authored-by: Lin <[email protected]>
  • Loading branch information
3 people authored and AntiZpvoh committed Jul 6, 2020
1 parent 94c2fcb commit 8534552
Show file tree
Hide file tree
Showing 9 changed files with 1 addition and 567 deletions.
2 changes: 1 addition & 1 deletion benchmark/opperf/utils/op_registry_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -52,7 +52,7 @@ def _select_ops(operator_names, filters=("_contrib", "_"), merge_op_forward_back
operators_with_backward = []

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

Expand Down
1 change: 0 additions & 1 deletion python/mxnet/contrib/amp/lists/symbol_bf16.py
Original file line number Diff line number Diff line change
Expand Up @@ -57,7 +57,6 @@
FP32_FUNCS = [
'Deconvolution',
'RNN',
'BatchNorm_v1',
'BilinearSampler',
'BlockGrad',
'Cast',
Expand Down
1 change: 0 additions & 1 deletion python/mxnet/contrib/amp/lists/symbol_fp16.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,7 +32,6 @@
# are dtype neutral (can work in both fp16 and fp32)
FP16_FP32_FUNCS = [
'BatchNorm',
'BatchNorm_v1',
'BilinearSampler',
'BlockGrad',
'Cast',
Expand Down
Loading

0 comments on commit 8534552

Please sign in to comment.