核 C++ API - 3.5 简体中文

Vitis AI 用户指南 (UG1414)

Document ID
UG1414
Release Date
2023-09-28
Version
3.5 简体中文

wego_torch::core::Compile

原型设计

torch::jit::Module Compile(const torch::jit::Module &module,
                           CompileOptions options);
表 1. 参数
类型 名称 描述
const torch::jit::Module & module 要编译的 PyTorch 量化模块,呈现为 torch::jit::Module 对象。
CompileOptions options 用于 WeGO-Torch 编译的编译器选项,指定为 wego_torch::core::CompileOptions 数据类型。如需获取有关此对象类的更多详细信息,请参阅核类章节。

返回

经最优化的 torch::jit::Module 对象。

wego_torch::core::GetTargetInfo

原型设计

TargetInfo GetTargetInfo();

参数

返回

wego_torch::TargetInfo 对象。请参阅各类部分,以获取有关此对象类的类型的更多详细信息。

wego_torch::getVersionInfo

原型设计

std::string getVersionInfo();

参数

返回

表示 wego_torch 版本信息的原始字符串。