|
|
@@ -147,8 +147,8 @@ def tf_installation_validation(func): |
|
|
type, inner function. |
|
|
type, inner function. |
|
|
""" |
|
|
""" |
|
|
|
|
|
|
|
|
def _f(graph_path: str, sample_shape: tuple, output_folder: str, report_folder: str = None, |
|
|
|
|
|
input_nodes: str = None, output_nodes: str = None): |
|
|
|
|
|
|
|
|
def _f(graph_path: str, input_nodes: dict, output_nodes: List[str], |
|
|
|
|
|
output_folder: str, report_folder: str): |
|
|
not_integral_error = RuntimeIntegrityError( |
|
|
not_integral_error = RuntimeIntegrityError( |
|
|
f"TensorFlow, " |
|
|
f"TensorFlow, " |
|
|
f"{get_third_part_lib_validation_error_info(['tf2onnx', 'onnx', 'onnxruntime', 'onnxoptimizer'])} " |
|
|
f"{get_third_part_lib_validation_error_info(['tf2onnx', 'onnx', 'onnxruntime', 'onnxoptimizer'])} " |
|
|
@@ -171,9 +171,9 @@ def tf_installation_validation(func): |
|
|
_print_error(not_integral_error) |
|
|
_print_error(not_integral_error) |
|
|
sys.exit(0) |
|
|
sys.exit(0) |
|
|
|
|
|
|
|
|
func(graph_path=graph_path, sample_shape=sample_shape, |
|
|
|
|
|
output_folder=output_folder, report_folder=report_folder, |
|
|
|
|
|
input_nodes=input_nodes, output_nodes=output_nodes) |
|
|
|
|
|
|
|
|
func(graph_path=graph_path, |
|
|
|
|
|
input_nodes=input_nodes, output_nodes=output_nodes, |
|
|
|
|
|
output_folder=output_folder, report_folder=report_folder) |
|
|
|
|
|
|
|
|
return _f |
|
|
return _f |
|
|
|
|
|
|
|
|
|