函式 torch_tensorrt::torchscript::check_method_operator_support

• 定義於 檔案 torch_tensorrt.h

函式文件

TORCHTRT_API bool torch_tensorrt::torchscript::check_method_operator_support(const torch::jit::Module &module, std::string method_name)

檢查模組是否完全受編譯器支援。

接受一個模組和一個方法名稱，檢查方法圖是否包含純粹可轉換的運算子

如果圖不受支援，將打印出不受支援的運算子列表

引數

• module – torch::jit::script::Module - 現有的 TorchScript 模組

• method_name – std::string - 要編譯的方法名稱

返回值

bool: 方法受 Torch-TensorRT.TorchScript 支援