megenginelite.struct¶
- 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¶