Function trtorch::CheckMethodOperatorSupport ¶
-
Defined in File trtorch.h
Function Documentation ¶
-
TRTORCH_API bool
trtorch
::
CheckMethodOperatorSupport
( 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 TRTorch