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

Add reference to options cpp

Joel Schlosser
2021-07-22 13:25:42 -04:00
parent 3ca850e95d
commit a6871a9ce4
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
Module-Onboarding-Lab.md

@ -34,7 +34,7 @@ each of the wrappers above.
* `torch/nn/modules/__init__.py` - add the module class name to the list in `__all__`
* `torch/nn/functional.py` - add the Python functional form of the module here
* `torch/csrc/api/include/torch/nn/modules/linear.h` / `torch/csrc/api/src/nn/modules/linear.cpp` - add the C++ module class definitions here
* `torch/csrc/api/include/torch/nn/options/linear.h` - add options for the module here (i.e. `num_features`)
* `torch/csrc/api/include/torch/nn/options/linear.h` / `torch/csrc/api/src/nn/options/linear.cpp` - add options for the module here (i.e. `num_features`)
* `torch/testing/_internal/common_nn.py` - add a `dict` entry to the `module_tests` list defining how to generate generic tests for the new module (note: this part is in the process of being improved)
* `test/test_nn.py` - add any module functionality tests here as functions on `TestNN` or `TestNNDeviceType` (for device-specific tests)
* `test/cpp/api/functional.cpp` - add C++ API functional tests here