wego_torch::core::Compile
Prototype
torch::jit::Module Compile(const torch::jit::Module &module,
CompileOptions options);
Type | Name | Description |
---|---|---|
const torch::jit::Module & | module | A quantized PyTorch module to be compiled, represented as a torch::jit::Module object. |
CompileOptions | options | Compiler options used for WeGO-Torch compilation, specified as the wego_torch::core::CompileOptions data type. For further details on this object class, refer to the core classes section. |
Return
An optimized torch::jit::Module object.
wego_torch::core::GetTargetInfo
Prototype
TargetInfo GetTargetInfo();
Parameters
None
Return
A wego_torch::TargetInfo object. See the classes section for more details about this object class type.
wego_torch::getVersionInfo
Prototype
std::string getVersionInfo();
Parameters
None
Return
A raw string representing wego_torch version information.