Function torch_tensorrt::torchscript::check_method_operator_support ¶
-
Defined in File torch_tensorrt.h
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
- Parameters
-
-
module
: torch::jit::script::Module - Existing TorchScript module -
method_name
: std::string - Name of method to compile
-
Will print out a list of unsupported operators if the graph is unsupported
- Return
-
bool: Method is supported by Torch-TensorRT.TorchScript