Support Embedding + EmbeddingBag in Script + (Ignore flakey test) (#14509)
authorElias Ellison <eellison@fb.com>
Thu, 29 Nov 2018 03:14:16 +0000 (19:14 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 29 Nov 2018 03:16:38 +0000 (19:16 -0800)
commit6d63e9dbfffba9f925ac3af5232390a76aa54dce
tree7924d57fc8f42429c12a911e3234c7ce49cbd991
parent105fa58748076a19682a4c0c9dee878a9575d7ed
Support Embedding + EmbeddingBag in Script + (Ignore flakey test) (#14509)

Summary:
Resubmitting PR #14415

The tests added for Embedding + EmbeddingBag had random numbers as input, which affected the random number generator & caused the flakey test to break.

Everything but the last two commits have already been accepted
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14509

Differential Revision: D13247917

Pulled By: eellison

fbshipit-source-id: ea6963c47f666c07687787e2fa82020cddc6aa15
aten/src/ATen/core/aten_interned_strings.h
aten/src/ATen/native/Embedding.cpp
aten/src/ATen/native/cuda/Embedding.cu
aten/src/ATen/native/native_functions.yaml
test/test_jit.py
tools/autograd/derivatives.yaml
torch/csrc/jit/script/init.cpp
torch/jit/__init__.py
torch/nn/functional.py
torch/nn/modules/sparse.py