projects / platform / upstream / pytorch.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ba25b37) | patch
Use nn module tests in test_jit (#14238)
authorDavid Riazati <davidriazati@fb.com>
Thu, 29 Nov 2018 07:28:59 +0000 (23:28 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Thu, 29 Nov 2018 07:31:25 +0000 (23:31 -0800)
commit9e93a026247c5a09c7f71ed3e36a9273c5477c73
tree6af870ed4cc04301627ff4df07b6dccedb338652tree | snapshot
parentba25b37e9b254f33990b1ea1399f8b396f9b47c7commit | diff
Use nn module tests in test_jit (#14238)

Summary:
This PR adds weak modules for all activation modules and uses `test_nn` module tests to test weak modules that have been annotated with `weak_module` and therefore are in `torch._jit_internal._weak_types`

Also depends on #14379
Pull Request resolved: https://github.com/pytorch/pytorch/pull/14238

Differential Revision: D13252887

Pulled By: driazati

fbshipit-source-id: e9638cf74089884a32b8f0f38396cf432c02c988
test/common_nn.py diff | blob | history
test/expect/TestJit.test_alexnet.expect diff | blob | history
test/test_jit.py diff | blob | history
test/test_nn.py diff | blob | history
torch/nn/functional.py diff | blob | history
torch/nn/modules/activation.py diff | blob | history
Domain: Machine Learning / ML Framework;