Implementation of WeightedSum op for mkl-dnn and fix FC op output shape issue.
authorPenghuiCheng <42089598+penghuicheng@users.noreply.github.com>
Fri, 7 Dec 2018 20:01:44 +0000 (12:01 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Fri, 7 Dec 2018 20:35:19 +0000 (12:35 -0800)
commit939877bf4b5f37307eebd0cb035d65212b9449f6
treecc0d9c9e1998512728036eed5f9b2b31bf4cf502
parent265b55d02845b91d6c8f94dabde69ebc0b0e7f46
Implementation of WeightedSum op for mkl-dnn and fix FC op output shape issue.

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

Reviewed By: yinghai

Differential Revision: D13364364

Pulled By: wesolwsk

fbshipit-source-id: e69bcd1bc52e35b2f0e45e5dc40184f1bd66605d
caffe2/ideep/operators/fully_connected_op.cc
caffe2/ideep/operators/operator_fallback_ideep.cc
caffe2/ideep/operators/utility_ops.cc
caffe2/python/ideep/fc_op_test.py
caffe2/python/ideep/weightedsum_op_test.py [new file with mode: 0644]