Resubmit: Set the correct engine name for position weighted pooling when fp16 is...
authorJiyan Yang <chocjy@fb.com>
Tue, 27 Nov 2018 22:49:28 +0000 (14:49 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Tue, 27 Nov 2018 22:51:56 +0000 (14:51 -0800)
commit0199d59d3a3ba69c62480ddb6680c9c5f55667b6
treec65d5821a2a05feee385500de916fa1b0d99c7c1
parentae1b37650c039b4b801299b9ab8b7beb4e727553
Resubmit: Set the correct engine name for position weighted pooling when fp16 is used for training

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

Reviewed By: xianjiec

Differential Revision: D12996103

fbshipit-source-id: 5ca4cda4210f68ece2b5d6eced8cf52ee91fb36f
caffe2/python/layers/sparse_lookup.py