Function trtorch::CheckMethodOperatorSupport

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