!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. MindSpore Lite Python API.
""" """
from __future__ import absolute_import
from .context import Context, DeviceInfo, CPUDeviceInfo, GPUDeviceInfo, AscendDeviceInfo from mindspore_lite.version import __version__
from .converter import FmkType, Converter from mindspore_lite.context import Context, DeviceInfo, CPUDeviceInfo, GPUDeviceInfo, AscendDeviceInfo
from .model import ModelType, Model, RunnerConfig, ModelParallelRunner from mindspore_lite.converter import FmkType, Converter
from .tensor import DataType, Format, Tensor from mindspore_lite.model import ModelType, Model, RunnerConfig, ModelParallelRunner
from mindspore_lite.tensor import DataType, Format, Tensor
__all__ = [] __all__ = []
__all__.extend(__version__)
__all__.extend(context.__all__) __all__.extend(context.__all__)
__all__.extend(converter.__all__) __all__.extend(converter.__all__)
__all__.extend(model.__all__) __all__.extend(model.__all__)

View File

@ -15,8 +15,9 @@
""" """
Context API. Context API.
""" """
from ._checkparam import check_isinstance, check_list_of_element from __future__ import absolute_import
from .lib import _c_lite_wrapper 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'] __all__ = ['Context', 'DeviceInfo', 'CPUDeviceInfo', 'GPUDeviceInfo', 'AscendDeviceInfo']

View File

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

View File

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

View File

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