projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: c904416) | patch
Pass IValue from c10 dispatcher to caffe2 operator (#16065)
authorSebastian Messmer <messmer@fb.com>
Fri, 18 Jan 2019 23:55:57 +0000 (15:55 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Sat, 19 Jan 2019 00:02:18 +0000 (16:02 -0800)
commite8b872abe225dfc9f622eaa847e82ad7296ea17b
treeae27181b414f5a1b1af0b25863689c8dd563048etree | snapshot
parentc9044166a5bf449dfbb56052235d6266a3abb5a1commit | diff
Pass IValue from c10 dispatcher to caffe2 operator (#16065)

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

Before, we registered the caffe2 kernel with the c10 dispatcher using plain C types.
Now, we pass in IValues, which avoids the unwrapping inbetween.

Reviewed By: ezyang

Differential Revision: D13689036

fbshipit-source-id: b976a2c46a5a541f6a926b3df255e8a535e32420
caffe2/operators/layer_norm_op.cc diff | blob | history
Domain: Machine Learning / ML Framework;