75 lines
3.3 KiB
Python
75 lines
3.3 KiB
Python
|
# type: ignore
|
||
|
|
from __future__ import annotations
|
||
|
|
from openvino._pyopenvino import Dimension
|
||
|
|
from openvino._pyopenvino import PartialShape
|
||
|
|
from openvino._pyopenvino import Type
|
||
|
|
from packaging.version import Version
|
||
|
|
from packaging.version import parse
|
||
|
|
import logging as log
|
||
|
|
import numpy as np
|
||
|
|
import openvino._pyopenvino
|
||
|
|
import sys as sys
|
||
|
|
__all__ = ['Dimension', 'PartialShape', 'Type', 'Version', 'create_generic_function_from_keras_model', 'create_tf_graph_iterator', 'extract_model_graph', 'get_concrete_func', 'get_environment_setup', 'get_imported_module_version', 'get_input_spec_from_model', 'get_signature_from_input', 'get_signature_from_input_signature', 'get_static_shape', 'is_variable', 'log', 'model_is_graph_iterator', 'np', 'parse', 'partial_shape_to_list', 'sys', 'tf_type_to_ov_type', 'trace_tf_model', 'trace_tf_model_if_needed', 'type_supported_by_tf_fe']
|
||
|
|
def create_generic_function_from_keras_model(keras_model):
|
||
|
|
...
|
||
|
|
def create_tf_graph_iterator(input_model, placeholder_shapes, placeholder_data_types, example_input, share_weights):
|
||
|
|
...
|
||
|
|
def extract_model_graph(argv):
|
||
|
|
...
|
||
|
|
def get_concrete_func(tf_function, example_input, input_needs_packing, error_message, use_example_input = True):
|
||
|
|
"""
|
||
|
|
|
||
|
|
Runs tracing of TF function and returns a concrete function.
|
||
|
|
|
||
|
|
:param tf_function: TF function that needs to be traced.
|
||
|
|
:param example_input: Example of function input.
|
||
|
|
:param input_needs_packing: determines if input needs to be packed in a list before passing to TF function.
|
||
|
|
It is used when original function was wrapped in outer TF function, which changes function signature.
|
||
|
|
In this case wrapper TF function always expects list of inputs which are unpacked inside subfunction.
|
||
|
|
So list/tuple are treated as multiple inputs of original model.
|
||
|
|
Non list/tuple are treated as single input, and it needs packing to a list,
|
||
|
|
as wrapper function always expect list of inputs.
|
||
|
|
:param error_message: Error message which should be shown in case of tracing error.
|
||
|
|
:param use_example_input: Determines if example_input should be used.
|
||
|
|
|
||
|
|
:returns: Object of type tf.types.experimental.ConcreteFunction.
|
||
|
|
|
||
|
|
"""
|
||
|
|
def get_environment_setup(framework):
|
||
|
|
"""
|
||
|
|
|
||
|
|
Get environment setup such as Python version, TensorFlow version
|
||
|
|
:param framework: framework name
|
||
|
|
:return: a dictionary of environment variables
|
||
|
|
|
||
|
|
"""
|
||
|
|
def get_imported_module_version(imported_module):
|
||
|
|
"""
|
||
|
|
|
||
|
|
Get imported module version
|
||
|
|
:return: version(str) or raise AttributeError exception
|
||
|
|
|
||
|
|
"""
|
||
|
|
def get_input_spec_from_model(model, input_shapes = None):
|
||
|
|
...
|
||
|
|
def get_signature_from_input(keras_model):
|
||
|
|
...
|
||
|
|
def get_signature_from_input_signature(keras_model):
|
||
|
|
...
|
||
|
|
def get_static_shape(shape: [openvino._pyopenvino.PartialShape, list, tuple], dynamic_value = None):
|
||
|
|
...
|
||
|
|
def is_variable(func_input, captures):
|
||
|
|
...
|
||
|
|
def model_is_graph_iterator(model):
|
||
|
|
...
|
||
|
|
def partial_shape_to_list(partial_shape: openvino._pyopenvino.PartialShape):
|
||
|
|
...
|
||
|
|
def tf_type_to_ov_type(val):
|
||
|
|
...
|
||
|
|
def trace_tf_model(model, input_shapes, input_types, example_input):
|
||
|
|
...
|
||
|
|
def trace_tf_model_if_needed(input_model, placeholder_shapes, placeholder_data_types, example_input):
|
||
|
|
...
|
||
|
|
def type_supported_by_tf_fe(input_model):
|
||
|
|
...
|