Skip to content

Commit

Permalink
docs: remove linear
Browse files Browse the repository at this point in the history
- also add poster pdf (was not public before, only png on icml site)
  • Loading branch information
plutonium-239 committed Aug 29, 2024
1 parent c41e06f commit 0327bc2
Show file tree
Hide file tree
Showing 4 changed files with 1 addition and 5 deletions.
2 changes: 0 additions & 2 deletions docs_src/api/nn/functional/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -12,13 +12,11 @@ This module contains the following members:
* :func:`batch_normMemSave`
* :func:`convMemSave`
* :func:`dropoutMemSave`
* :func:`linearMemSave`
* :func:`maxpool2dMemSave`
* :func:`reluMemSave`

.. autofunction:: batch_normMemSave
.. autofunction:: convMemSave
.. autofunction:: dropoutMemSave
.. autofunction:: linearMemSave
.. autofunction:: maxpool2dMemSave
.. autofunction:: reluMemSave
2 changes: 0 additions & 2 deletions docs_src/api/nn/index.rst
Original file line number Diff line number Diff line change
Expand Up @@ -11,7 +11,6 @@ This module contains the following members:
* :class:`MemSaveConvTranspose1d <memsave_torch.nn.MemSaveConvTranspose2d>`
* :class:`MemSaveConvTranspose2d <memsave_torch.nn.MemSaveConvTranspose2d>`
* :class:`MemSaveConvTranspose3d <memsave_torch.nn.MemSaveConvTranspose2d>`
* :class:`MemSaveLinear <memsave_torch.nn.MemSaveLinear>`
* :class:`MemSaveReLU <memsave_torch.nn.MemSaveReLU>`
* :class:`MemSaveMaxPool2d <memsave_torch.nn.MemSaveMaxPool2d>`
* :class:`MemSaveBatchNorm2d <memsave_torch.nn.MemSaveBatchNorm2d>`
Expand Down Expand Up @@ -41,7 +40,6 @@ Learnable Layers
MemSaveConvTranspose1d
MemSaveConvTranspose2d
MemSaveConvTranspose3d
MemSaveLinear


Activations and Pooling Layers
Expand Down
2 changes: 1 addition & 1 deletion docs_src/conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -64,7 +64,7 @@
"github_url": "https://github.com/plutonium-239/memsave_torch",
"og_image_url": "https://memsave-torch.readthedocs.io/_static/memsave_torch_logo_inv_text_256x256.png",
"nav_links": [
{"title": "memsave_torch", "url": "/index"},
{"title": "memsave_torch", "url": "index"},
],
"nav_links_align": "center",
}
Expand Down
Binary file added memsave_poster.pdf
Binary file not shown.

0 comments on commit 0327bc2

Please sign in to comment.