projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 64186e0) | patch
Introduce backend extensions (overriding operators on custom backends)
authorRoy Li <royboy@fb.com>
Fri, 1 Feb 2019 18:55:00 +0000 (10:55 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 1 Feb 2019 19:00:16 +0000 (11:00 -0800)
commit7e642dfff3a4bbf80826f331312acff2672a3e02
tree2b8499a1824bf467cfb152ec1089be517bcae520tree | snapshot
parent64186e06eccad8721cfa741cae31a39842a3dfe3commit | diff
Introduce backend extensions (overriding operators on custom backends)

Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/15153

Reviewed By: gchanan

Differential Revision: D13445571

fbshipit-source-id: 62e2ebe0a6e81c4983b47cddb57ee5eb78e96708
17 files changed:
aten/src/ATen/core/Type.h diff | blob | history
aten/src/ATen/function_wrapper.py diff | blob | history
aten/src/ATen/gen.py diff | blob | history
aten/src/ATen/templates/ExtensionBackendRegistration.h [new file with mode: 0644] blob
aten/src/ATen/templates/TypeExtension.cpp [new file with mode: 0644] blob
aten/src/ATen/templates/TypeExtension.h [new file with mode: 0644] blob
aten/src/ATen/test/CMakeLists.txt diff | blob | history
aten/src/ATen/test/extension_backend_test.cpp [new file with mode: 0644] blob
aten/tools/run_tests.sh diff | blob | history
c10/core/Backend.h diff | blob | history
c10/core/DeviceType.cpp diff | blob | history
c10/core/DeviceType.h diff | blob | history
c10/core/TensorImpl.h diff | blob | history
c10/core/TensorOptions.h diff | blob | history
c10/core/TensorTypeIdRegistration.cpp diff | blob | history
c10/core/TensorTypeIdRegistration.h diff | blob | history
caffe2/proto/caffe2.proto diff | blob | history
Domain: Machine Learning / ML Framework;