Skip to content

Commit

Permalink
Merge branch 'dl/conv_layer_attrs_update' into dl/quantization/passes…
Browse files Browse the repository at this point in the history
…_for_splitted_graphs
  • Loading branch information
daniil-lyakhov committed Nov 16, 2023
2 parents 6bc2f86 + 3fb5f9a commit d27a7f3
Showing 1 changed file with 4 additions and 7 deletions.
11 changes: 4 additions & 7 deletions nncf/openvino/graph/layer_attributes.py
Original file line number Diff line number Diff line change
Expand Up @@ -80,10 +80,7 @@ def get_backend_agnostic_attributes(self):
]


OVConvLayout = List[OVLayoutElem]


def get_conv_weights_layout_from_node(node: NNCFNode) -> OVConvLayout:
def get_conv_weights_layout_from_node(node: NNCFNode) -> List[OVLayoutElem]:
"""
Calculates weights layout for a target convolution node.
Expand All @@ -97,7 +94,7 @@ def get_conv_weights_layout_from_node(node: NNCFNode) -> OVConvLayout:
)


def get_linear_weights_layout_from_node(node: NNCFNode) -> OVConvLayout:
def get_linear_weights_layout_from_node(node: NNCFNode) -> List[OVLayoutElem]:
"""
Calculates weights layout for a target linear node.
Expand Down Expand Up @@ -126,7 +123,7 @@ def _get_constant_port_id_from_layer_attributes(layer_attributes: OVLayerAttribu
return port_ids[0]


def get_conv_weights_layout(ov_metatype: OVOpMetatype, weights_shape: Tuple[int, ...]) -> OVConvLayout:
def get_conv_weights_layout(ov_metatype: OVOpMetatype, weights_shape: Tuple[int, ...]) -> List[OVLayoutElem]:
"""
Calculates weights layout for a target convolution node.
Expand All @@ -140,7 +137,7 @@ def get_conv_weights_layout(ov_metatype: OVOpMetatype, weights_shape: Tuple[int,
return tuple(weights_layout)


def get_linear_weights_layout(weights_shape: Tuple[int, ...], transpose: bool, port_id: int) -> OVConvLayout:
def get_linear_weights_layout(weights_shape: Tuple[int, ...], transpose: bool, port_id: int) -> List[OVLayoutElem]:
"""
Calculates weights layout for a target linear node.
Expand Down

0 comments on commit d27a7f3

Please sign in to comment.