frozenleaves/pytorch
1
0
Fork 0
mirror of https://github.com/pytorch/pytorch.git synced 2025-10-20 21:14:14 +08:00
Code Issues Packages Projects Releases Wiki Activity

Python/C++ API Parity: Add impl and tests for ParameterDict (#40654)

Browse Source 
Summary:
This diff contains the implementation of C++ api for ParameterDict from https://github.com/pytorch/pytorch/issues/25883, refer to  https://github.com/pytorch/pytorch/issues/36904 and https://github.com/pytorch/pytorch/issues/28652
Pull Request resolved: https://github.com/pytorch/pytorch/pull/40654

Test Plan: Add unit test in this diff

Differential Revision: D22273265

Pulled By: glaringlee

fbshipit-source-id: 9134a92c95eacdd53d5b24470d5f7edbeb40a488
This commit is contained in:
yyn19951228
2020-06-29 08:48:22 -07:00
committed by Facebook GitHub Bot
parent b35cdc5200
commit 4121d34036
5 changed files with 297 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
test/cpp/api/CMakeLists.txt
View File

@ -15,6 +15,7 @@ set(TORCH_API_TEST_SOURCES
${TORCH_API_TEST_DIR}/module.cpp
${TORCH_API_TEST_DIR}/modulelist.cpp
${TORCH_API_TEST_DIR}/modules.cpp
${TORCH_API_TEST_DIR}/parameterdict.cpp
${TORCH_API_TEST_DIR}/namespace.cpp
${TORCH_API_TEST_DIR}/nn_utils.cpp
${TORCH_API_TEST_DIR}/optim.cpp