grid_sample should be fp32 for now

656d14b0 · Michael Carilli · 9ce80178 · 656d14b0
Commit 656d14b0 authored Jun 25, 2019 by Michael Carilli
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 0 deletions

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

No files found.
--- a/apex/amp/lists/functional_overrides.py
+++ b/apex/amp/lists/functional_overrides.py
@@ -30,6 +30,7 @@ FP32_FUNCS = [

    # Interpolation/Upsampling TODO:  Remove for 1.2
    'interpolate',
+    'grid_sample',

    # Pointwise
    'softplus',