25 lines
1.1 KiB
Python
25 lines
1.1 KiB
Python
# type: ignore
|
|
from __future__ import annotations
|
|
from collections.abc import Callable
|
|
from openvino._pyopenvino import PartialShape
|
|
from openvino.tools.ovc.error import Error
|
|
from openvino.tools.ovc.utils import refer_to_faq_msg
|
|
import collections.abc
|
|
import openvino._pyopenvino
|
|
__all__ = ['Callable', 'Error', 'PartialShape', 'get_dimension_index_by_label', 'refer_to_faq_msg', 'update_layout_to_dict']
|
|
def get_dimension_index_by_label(input_shape: openvino._pyopenvino.PartialShape, input_names: list, layout_dict: [dict], dimension_label: str, default_dim: int):
|
|
"""
|
|
|
|
The function returns index of the dimension pointed in the layout
|
|
and a flag indicating if the index is chosen by default.
|
|
For example, the index for 'D' dimension in "NHWDC" layout is 3.
|
|
|
|
"""
|
|
def update_layout_to_dict(inputs: list, layout: [list, dict], get_names_func: collections.abc.Callable):
|
|
"""
|
|
|
|
The function prepares layout values in the dictionary with items of the format:
|
|
{ node_name : {'source_layout': 'NHWC', 'target_layout': 'NCHW'} }
|
|
|
|
"""
|