Skip to content

Commit 310e61b

Browse files
irenabirenab
authored andcommitted
restore network editor inports in __init__, remove quant funcions rules from docs
1 parent 98c3f2d commit 310e61b

2 files changed

Lines changed: 8 additions & 8 deletions

File tree

docsrc/source/api/api_docs/modules/network_editor.rst

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -42,14 +42,6 @@ Actions
4242

4343
|
4444
45-
.. autoclass:: model_compression_toolkit.core.network_editor.ChangeCandidatesActivationQuantConfigAttr
46-
47-
|
48-
49-
.. autoclass:: model_compression_toolkit.core.network_editor.ChangeQuantizationParamFunction
50-
51-
|
52-
5345
.. autoclass:: model_compression_toolkit.core.network_editor.ChangeFinalWeightsQuantizationMethod
5446

5547
|

model_compression_toolkit/core/common/network_editors/__init__.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -13,6 +13,14 @@
1313
# limitations under the License.
1414
# ==============================================================================
1515

16+
from model_compression_toolkit.core.common.network_editors.actions import (
17+
ChangeCandidatesWeightsQuantConfigAttr,
18+
ChangeFinalWeightsQuantConfigAttr,
19+
ChangeCandidatesActivationQuantConfigAttr,
20+
ChangeCandidatesActivationQuantizationMethod,
21+
ChangeFinalWeightsQuantizationMethod,
22+
ChangeCandidatesWeightsQuantizationMethod,
23+
ChangeFinalActivationQuantConfigAttr)
1624
from model_compression_toolkit.core.common.network_editors.actions import EditRule
1725
from model_compression_toolkit.core.common.network_editors.node_filters import NodeTypeFilter, NodeNameScopeFilter, \
1826
NodeNameFilter

0 commit comments

Comments
 (0)