函数 torch_tensorrt::torchscript::check_method_operator_support

函数文档

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 支持。