megenginelite.struct#

class LiteBackend(value)[源代码]#

推理后端枚举类型,目前仅有 DEFAULT

LITE_DEFAULT = 0#
class LiteDeviceType(value)[源代码]#

后端设备枚举类型

备注

计算和存储都在盖设备上

LITE_ATLAS = 3#
LITE_CAMBRICON = 5#
LITE_CPU = 0#
LITE_CUDA = 1#
LITE_DEVICE_DEFAULT = 7#
LITE_NPU = 4#
class LiteDataType(value)[源代码]#

tensor 数据类型枚举类型

备注

half表示float16, int表示int32

LITE_FLOAT = 0#
LITE_HALF = 1#
LITE_INT = 2#
LITE_INT16 = 3#
LITE_INT8 = 4#
LITE_UINT16 = 6#
LITE_UINT8 = 5#
class LiteTensorPhase(value)[源代码]#

tesnor 类型枚举类型 .. note:: LITE_IO既可以表示LITE_INPUT也可以表示LITE_OUTPUT

LITE_INPUT = 1#
LITE_IO = 0#
LITE_OUTPUT = 2#
class LiteIOType(value)[源代码]#

输入输出类型枚举,包括SHAPE和VALUE, 有时用户仅仅需要输入tensor的shape

LITE_IO_SHAPE = 1#
LITE_IO_VALUE = 0#
class LiteAlgoSelectStrategy(value)[源代码]#

算子算法选择策略枚举,部分算子实现有多种算法,不同算法有不同的属性,根据策略选择最优的算法

备注

LITE_ALGO_HEURISTIC | LITE_ALGO_PROFILE 的策略组合表示如果profile的算法缓存无法获取则使用启发式方案替代

LITE_ALGO_HEURISTIC | LITE_ALGO_REPRODUCIBLE 表示启发式选择可重入的算法

LITE_ALGO_PROFILE | LITE_ALGO_REPRODUCIBLE表示从可重入的算法集合中评估获取最优的算法

LITE_ALGO_PROFILE | LITE_ALGO_OPTIMIZED表示从优化的算法中评估获取最优的算法,评估过程可以快速完成

LITE_ALGO_PROFILE | LITE_ALGO_OPTIMIZED | LITE_ALGO_REPRODUCIBLE表示从优化的可重入算法中评估选择最优算法

LITE_ALGO_HEURISTIC = 1#
LITE_ALGO_OPTIMIZED = 8#
LITE_ALGO_PROFILE = 2#
LITE_ALGO_REPRODUCIBLE = 4#
class LiteLogLevel(value)[源代码]#

日志等级枚举类型

备注

DEBUG: 最详细的等级,会打印debug相关信息

INFO: 默认等级

WARN:打印警告信息

ERROR:最简略的等级,只打印错误信息

DEBUG = 0#
ERROR = 3#
INFO = 1#
WARN = 2#