Function torch_tensorrt::torchscript::check_method_operator_support¶
Function Documentation¶
-
TORCHTRT_API bool torch_tensorrt::torchscript::check_method_operator_support(const torch::jit::Module &module, std::string method_name)¶
Check to see if a module is fully supported by the compiler.
Takes a module and a method name and checks if the method graph contains purely convertable operators
Will print out a list of unsupported operators if the graph is unsupported
- Parameters
module, : – torch::jit::script::Module - Existing TorchScript module
method_name, : – std::string - Name of method to compile
- Returns
bool: Method is supported by Torch-TensorRT.TorchScript
