Validate matching input shapes in Int8Add operator (#14520)
authorMarat Dukhan <marat@fb.com>
Wed, 5 Dec 2018 19:39:46 +0000 (11:39 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Wed, 5 Dec 2018 20:00:23 +0000 (12:00 -0800)
commit55092b1cc604fad3d70d31e71bbdd3a43a279423
tree63a8428b29b37b9904bf1aabd9a6e5217d38ea48
parent1c2273c8e99730b4684c06daebe5528adeeb0155
Validate matching input shapes in Int8Add operator (#14520)

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

Default engine doesn't support broadcast semantics in Int8Add operator. This patch adds a check that shapes are equivalent.

Reviewed By: bertmaher

Differential Revision: D13250922

fbshipit-source-id: 8526d07723bd9a34d54dee04d121c57f8b33c481
caffe2/operators/quantized/int8_add_op.h