函数 torch_tensorrt::torchscript::check_method_operator_support

• 定义于 文件 torch_tensorrt.h

函数文档

TORCHTRT_API bool torch_tensorrt::torchscript::check_method_operator_support(const torch::jit::Module &module, std::string method_name)

检查模块是否完全受编译器支持。

接收一个模块和一个方法名，并检查该方法图是否仅包含可转换的操作符

如果图不受支持，将打印出不支持的操作符列表

参数

• module – torch::jit::script::Module - 现有的 TorchScript 模块

• method_name – std::string - 要编译的方法名称

返回

bool: 方法是否受 Torch-TensorRT.TorchScript 支持