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
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