Function torch::jit::import_ir_module(std::shared_ptr<CompilationUnit>, std::shared_ptr<caffe2::serialize::PyTorchStreamReader>, std::shared_ptr<torch::jit::DeserializationStorageContext>, c10::optional<at::Device>, 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, c10::optional<at::Device> device, std::string ts_id)