Skip to main content

XMOS 人工智能工具

项目描述

XMOS 人工智能工具

用法

使用 xformer

from xmos_ai_tools import xformer as xf

xf.convert("source model path", "converted model path", params=None)

whereparams是编译器标志和参数及其值的字典。

例如:

from xmos_ai_tools import xformer as xf

xf.convert("example_int8_model.tflite", "xcore_optimised_example_int8_model.tflite", {
    "mlir-disable-threading": None,
    "xcore-reduce-memory": None,
})

要查看所有可用参数,请调用

from xmos_ai_tools import xformer as xf

xf.print_help()

这将打印所有可传递给 xformer 的选项。要查看隐藏选项,请运行print_help(show_hidden=True)

使用 xcore tflm 主机解释器

from xmos_ai_tools import xcore_tflm_host_interpreter as xtflm

ie = xtflm.XTFLMInterpreter(model_content=xformed_model)
ie.set_input_tensor(0, input_tensor)
ie.invoke()
xformer_outputs = []
for i in range(num_of_outputs):
    xformer_outputs.append(ie.get_output_tensor(i))

下载文件

下载适用于您平台的文件。如果您不确定要选择哪个,请了解有关安装包的更多信息。

内置发行版