projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ac4f66c) | patch
C10_REGISTER_CAFFE2_OPERATOR: Macro for registering c2 kernels (#16548)
authorSebastian Messmer <messmer@fb.com>
Thu, 7 Feb 2019 21:52:49 +0000 (13:52 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 7 Feb 2019 21:58:14 +0000 (13:58 -0800)
commit6750e1e3e995f2070464236d1585a772e647e524
treec81bc038b1cfb94b25d352dec432c3ca4dbf66f1tree | snapshot
parentac4f66c9c36525de965671387fc50f8feb101fa3commit | diff
C10_REGISTER_CAFFE2_OPERATOR: Macro for registering c2 kernels (#16548)

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

With this macro, a caffe2 operator can now directly be registered with c10.
No need to write custom wrapper kernels anymore.

Differential Revision: D13877076

fbshipit-source-id: e56846238c5bb4b1989b79855fd44d5ecf089c9c
aten/src/ATen/core/dispatch/DispatchTable.h diff | blob | history
aten/src/ATen/core/stack.h diff | blob | history
c10/util/flat_hash_map.h diff | blob | history
caffe2/core/c10_operator.h [new file with mode: 0644] blob
caffe2/core/operator.h diff | blob | history
caffe2/operators/layer_norm_op.cc diff | blob | history
caffe2/operators/layer_norm_op.h diff | blob | history
caffe2/python/operator_test/layer_norm_op_test.py diff | blob | history
test/test_torch.py diff | blob | history
Domain: Machine Learning / ML Framework;