check_method_operator_support

函数功能

检查TorchScript模型是否有不支持的算子,不支持的算子会在日志中以info级别输出。

函数原型

bool check_method_operator_support(const torch::jit::Module& module, const std::string& methodName);

约束说明

check_method_operator_support接口由于存在参数校验,在输入非法数据时,可能会抛出异常。故用户必须在try/catch语句块内进行调用以及异常处理,防止在使用的过程中出现异常抛出导致程序退出的情况。

参数说明

参数名

输入/输出

说明

module

输入

原始TorchScript模型。

methodName

输入

函数名称。String的长度范围为(0,1024],且为module中正确合法的函数名称。

返回值说明

算子全部支持返回true,否则返回false。