| 39 | } |
| 40 | |
| 41 | Expect<ErrNo> TorchScript::loadFromPath(const std::string &Path, |
| 42 | Device Device) { |
| 43 | if (auto Err = setDevice(Device); Err != ErrNo::Success) { |
| 44 | return Err; |
| 45 | } |
| 46 | TorchModel = torch::jit::load(Path); |
| 47 | return ErrNo::Success; |
| 48 | } |
| 49 | |
| 50 | Expect<ErrNo> TorchScript::run(std::vector<at::Tensor> In, |
| 51 | std::vector<at::Tensor> &Out) { |