函数 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)

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

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

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

Parameters

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

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

Returns

bool: 方法由 Torch-TensorRT.TorchScript 支持