review.tizen.org Git - platform/upstream/pytorch.git/commit

projects / platform / upstream / pytorch.git / commit

author	Elias Ellison <eellison@fb.com>
	Wed, 28 Nov 2018 18:50:26 +0000 (10:50 -0800)
committer	Facebook Github Bot <facebook-github-bot@users.noreply.github.com>
	Wed, 28 Nov 2018 18:52:30 +0000 (10:52 -0800)
commit	7749804099b8a64aea4bf91e298a20976f9b10ad
tree	290de4c378fdba73e94a1d13e67e8adcf9165abf	tree \| snapshot
parent	c32debb9165b46c3f10a42e0465e7d8e23ed11db	commit \| diff

Support Embedding + EmbeddingBag in Script (#14415)

Summary:
Add support for Embedding and EmbeddingBag in script. Both functions require with torch.no_grad(), which we don't have any plans to support in the near future. To work around this, I added a embedding_renorm function without derivatives.
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14415

Reviewed By: wanchaol

Differential Revision: D13219647

Pulled By: eellison

fbshipit-source-id: c90706aa6fbd48686eb10f3efdb65844be7b8717

Domain: Machine Learning / ML Framework;

RSS Atom

aten/src/ATen/core/aten_interned_strings.h		diff \| blob \| history
aten/src/ATen/native/Embedding.cpp		diff \| blob \| history
aten/src/ATen/native/cuda/Embedding.cu		diff \| blob \| history
aten/src/ATen/native/native_functions.yaml		diff \| blob \| history
test/test_jit.py		diff \| blob \| history
tools/autograd/derivatives.yaml		diff \| blob \| history
torch/csrc/jit/script/init.cpp		diff \| blob \| history
torch/jit/__init__.py		diff \| blob \| history
torch/nn/functional.py		diff \| blob \| history
torch/nn/modules/sparse.py		diff \| blob \| history