From abd4d64df16ae7c261fa2837062f551d856a8f96 Mon Sep 17 00:00:00 2001 From: Joel Schlosser <75754324+jbschlosser@users.noreply.github.com> Date: Fri, 27 Aug 2021 10:32:32 -0400 Subject: [PATCH] Updated Module Onboarding Lab (markdown) --- Module-Onboarding-Lab.md | 1 + 1 file changed, 1 insertion(+) diff --git a/Module-Onboarding-Lab.md b/Module-Onboarding-Lab.md index 924e913..4594d40 100644 --- a/Module-Onboarding-Lab.md +++ b/Module-Onboarding-Lab.md @@ -33,6 +33,7 @@ each of the wrappers above. * `torch/nn/modules/linear.py` - add the Python module class definition here * `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/functional/linear.h` - add the C++ functional form 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` / `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)