!49073 [MS][LITE]add python api version

Merge pull request !49073 from luoyuan/add_python_version
This commit is contained in:
i-robot 2023-03-08 02:25:02 +00:00 committed by Gitee
commit 5ea0de458e
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
5 changed files with 23 additions and 16 deletions

View File

@ -15,13 +15,16 @@
"""
MindSpore Lite Python API.
"""
from __future__ import absolute_import
from .context import Context, DeviceInfo, CPUDeviceInfo, GPUDeviceInfo, AscendDeviceInfo
from .converter import FmkType, Converter
from .model import ModelType, Model, RunnerConfig, ModelParallelRunner
from .tensor import DataType, Format, Tensor
from mindspore_lite.version import __version__
from mindspore_lite.context import Context, DeviceInfo, CPUDeviceInfo, GPUDeviceInfo, AscendDeviceInfo
from mindspore_lite.converter import FmkType, Converter
from mindspore_lite.model import ModelType, Model, RunnerConfig, ModelParallelRunner
from mindspore_lite.tensor import DataType, Format, Tensor
__all__ = []
__all__.extend(__version__)
__all__.extend(context.__all__)
__all__.extend(converter.__all__)
__all__.extend(model.__all__)

View File

@ -15,8 +15,9 @@
"""
Context API.
"""
from ._checkparam import check_isinstance, check_list_of_element
from .lib import _c_lite_wrapper
from __future__ import absolute_import
from mindspore_lite._checkparam import check_isinstance, check_list_of_element
from mindspore_lite.lib import _c_lite_wrapper
__all__ = ['Context', 'DeviceInfo', 'CPUDeviceInfo', 'GPUDeviceInfo', 'AscendDeviceInfo']

View File

@ -15,14 +15,15 @@
"""
Converter API.
"""
from __future__ import absolute_import
import os
from enum import Enum
from ._checkparam import check_isinstance, check_input_shape, check_config_info
from .lib import _c_lite_wrapper
from .tensor import DataType, Format, data_type_py_cxx_map, data_type_cxx_py_map, format_py_cxx_map, format_cxx_py_map
from .model import ModelType, model_type_py_cxx_map, model_type_cxx_py_map
from mindspore_lite._checkparam import check_isinstance, check_input_shape, check_config_info
from mindspore_lite.lib import _c_lite_wrapper
from mindspore_lite.tensor import DataType, Format, data_type_py_cxx_map, data_type_cxx_py_map, format_py_cxx_map, \
format_cxx_py_map
from mindspore_lite.model import ModelType, model_type_py_cxx_map, model_type_cxx_py_map
__all__ = ['FmkType', 'Converter']

View File

@ -15,13 +15,14 @@
"""
Model API.
"""
from __future__ import absolute_import
import os
from enum import Enum
from ._checkparam import check_isinstance
from .context import Context
from .lib import _c_lite_wrapper
from .tensor import Tensor
from mindspore_lite._checkparam import check_isinstance
from mindspore_lite.context import Context
from mindspore_lite.lib import _c_lite_wrapper
from mindspore_lite.tensor import Tensor
__all__ = ['ModelType', 'Model', 'RunnerConfig', 'ModelParallelRunner']

View File

@ -15,11 +15,12 @@
"""
Tensor API.
"""
from __future__ import absolute_import
from enum import Enum
import numpy
from .lib import _c_lite_wrapper
from mindspore_lite.lib import _c_lite_wrapper
__all__ = ['DataType', 'Format', 'Tensor']