wego_torch::core::Compile
原型设计
torch::jit::Module Compile(const torch::jit::Module &module,
CompileOptions options);
类型 | 名称 | 描述 |
---|---|---|
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 版本信息的原始字符串。