mirror of
https://github.com/ml-explore/mlx.git
synced 2025-08-09 10:46:39 +08:00
dff4a3833f
* update module to check weights on load, also fix docs and reorganize tests * nits + rebase * a few more docs updates for Module * use manual module file * comment
34 lines
693 B
ReStructuredText
34 lines
693 B
ReStructuredText
{{ fullname | escape | underline}}
|
|
|
|
.. currentmodule:: {{ module }}
|
|
|
|
.. add toctree option to make autodoc generate the pages
|
|
|
|
.. autoclass:: {{ objname }}
|
|
|
|
{% block attributes %}
|
|
{% if attributes %}
|
|
.. rubric:: Attributes
|
|
|
|
.. autosummary::
|
|
:toctree: .
|
|
{% for item in attributes %}
|
|
~{{ fullname }}.{{ item }}
|
|
{%- endfor %}
|
|
{% endif %}
|
|
{% endblock %}
|
|
|
|
{% block methods %}
|
|
{% if methods %}
|
|
.. rubric:: Methods
|
|
|
|
.. autosummary::
|
|
:toctree: .
|
|
{% for item in methods %}
|
|
{%- if item not in inherited_members and item != '__init__' %}
|
|
~{{ fullname }}.{{ item }}
|
|
{%- endif -%}
|
|
{%- endfor %}
|
|
{% endif %}
|
|
{% endblock %}
|