mirror of
https://github.com/saymrwulf/onnxruntime.git
synced 2026-05-15 20:50:42 +00:00
e53422c6d0
Add runtime optimization support to ONNX -> ORT format conversion script. Replace `--optimization_level`, `--use_nnapi`, and `--use_coreml` with a new `--optimization_style` option.
46 lines
1.5 KiB
Python
46 lines
1.5 KiB
Python
# Copyright (c) Microsoft Corporation. All rights reserved.
|
|
# Licensed under the MIT License.
|
|
|
|
import pathlib
|
|
import typing
|
|
import os
|
|
|
|
|
|
def path_match_suffix_ignore_case(path: typing.Union[pathlib.Path, str], suffix: str) -> bool:
|
|
'''
|
|
Returns whether `path` ends in `suffix`, ignoring case.
|
|
'''
|
|
if not isinstance(path, str):
|
|
path = str(path)
|
|
return path.casefold().endswith(suffix.casefold())
|
|
|
|
|
|
def files_from_file_or_dir(file_or_dir_path: typing.Union[pathlib.Path, str],
|
|
predicate: typing.Callable[[pathlib.Path], bool] = lambda _: True) \
|
|
-> typing.List[pathlib.Path]:
|
|
'''
|
|
Gets the files in `file_or_dir_path` satisfying `predicate`.
|
|
If `file_or_dir_path` is a file, the single file is considered. Otherwise, all files in the directory are
|
|
considered.
|
|
:param file_or_dir_path: Path to a file or directory.
|
|
:param predicate: Predicate to determine if a file is included.
|
|
:return: A list of files.
|
|
'''
|
|
if not isinstance(file_or_dir_path, pathlib.Path):
|
|
file_or_dir_path = pathlib.Path(file_or_dir_path)
|
|
|
|
selected_files = []
|
|
|
|
def process_file(file_path: pathlib.Path):
|
|
if predicate(file_path):
|
|
selected_files.append(file_path)
|
|
|
|
if file_or_dir_path.is_dir():
|
|
for root, _, files in os.walk(file_or_dir_path):
|
|
for file in files:
|
|
file_path = pathlib.Path(root, file)
|
|
process_file(file_path)
|
|
else:
|
|
process_file(file_or_dir_path)
|
|
|
|
return selected_files
|