Function torch::jit::load(std::shared_ptr<caffe2::serialize::ReadAdapterInterface>, std::optional<c10::Device>, bool)

• Defined in File import.h

Function Documentation

Module torch::jit::load(std::shared_ptr<caffe2::serialize::ReadAdapterInterface> rai, std::optional<c10::Device> device = std::nullopt, bool load_debug_files = true)

Loads a serialized Module from the given shared_ptr rai.

The reader adapter, which is for customized input stream, must contain a serialized Module, exported either via ScriptModule.save() in Python or torch::jit::ExportModule in C++.