Skip to content

Commit 295211e

Browse files
Comments
1 parent c2cae20 commit 295211e

File tree

1 file changed

+1
-1
lines changed
  • src/nncf/quantization/algorithms/weight_compression

1 file changed

+1
-1
lines changed

src/nncf/quantization/algorithms/weight_compression/algorithm.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -1133,7 +1133,7 @@ def apply_with_parameters(
11331133
)
11341134
return transformed_model
11351135

1136-
def _get_activation_node_port_and_channel(self, node: NNCFNode, nncf_graph: NNCFGraph) -> tuple[NNCFNode, int]:
1136+
def _get_activation_node_port_and_channel(self, node: NNCFNode, nncf_graph: NNCFGraph) -> tuple[NNCFNode, int, int]:
11371137
"""
11381138
This method returns the activation layer, corresponding port id and channel axis for the given node.
11391139

0 commit comments

Comments
 (0)