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