Shortcuts

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

Docs

Access comprehensive developer documentation for PyTorch

View Docs

Tutorials

Get in-depth tutorials for beginners and advanced developers

View Tutorials

Resources

Find development resources and get your questions answered

View Resources