Move Affine grid to C++ (#14392)
authorElias Ellison <eellison@fb.com>
Wed, 28 Nov 2018 02:36:05 +0000 (18:36 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 28 Nov 2018 02:38:11 +0000 (18:38 -0800)
commit82175f31b47680a72ad7c00ef185d9a0c1ceb078
tree181f82593c1759786dcdc018e3cfa8f88961e72a
parent6f2307ba6ac802f4c4650cd21869e49d1845188d
Move Affine grid to C++ (#14392)

Summary:
Port AffineGrid to C++, because script does not support compiling Function classes.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14392

Differential Revision: D13219698

Pulled By: eellison

fbshipit-source-id: 3ddad8a84c72010b5a6c6f7f9712be614202faa6
aten/src/ATen/core/aten_interned_strings.h
aten/src/ATen/native/AffineGridGenerator.cpp [new file with mode: 0644]
aten/src/ATen/native/native_functions.yaml
test/test_jit.py
tools/autograd/derivatives.yaml
torch/csrc/jit/register_prim_ops.cpp
torch/csrc/jit/script/compiler.cpp
torch/jit/__init__.py
torch/nn/_functions/vision.py
torch/nn/functional.py