Function torch::jit::import_ir_module(std::shared_ptr<CompilationUnit>, std::shared_ptr<caffe2::serialize::PyTorchStreamReader>, std::shared_ptr<torch::jit::DeserializationStorageContext>, std::optional<at::Device>, const std::string&)

• Defined in File import.h

Function Documentation

Module torch::jit::import_ir_module(std::shared_ptr<CompilationUnit> cu, std::shared_ptr<caffe2::serialize::PyTorchStreamReader> reader, std::shared_ptr<torch::jit::DeserializationStorageContext> storage_context, std::optional<at::Device> device, const std::string &ts_id)