Merge branch 'master' of https://github.com/NVIDIA/apex

e9bbfa59 · Michael Carilli · 2eccdbd2 · dc8f400f · e9bbfa59
Commit e9bbfa59 authored Apr 04, 2019 by Michael Carilli
Hide whitespace changes
Inline Side-by-side

Showing with 4 additions and 0 deletions

apex/amp/lists/functional_overrides.py apex/amp/lists/functional_overrides.py +4 -0

No files found.
--- a/apex/amp/lists/functional_overrides.py
+++ b/apex/amp/lists/functional_overrides.py
@@ -27,6 +27,10 @@ FP16_FUNCS = [
 ]

 FP32_FUNCS = [
+
+    # Interpolation/Upsampling
+    'interpolate',
+
    # Pointwise
    'softplus',
    'softmin',