1 | #pragma once |
---|---|
2 | |
3 | #include <torch/csrc/distributed/rpc/message.h> |
4 | #include <torch/csrc/distributed/rpc/rpc_command_base.h> |
5 | #include <torch/csrc/jit/serialization/pickler.h> |
6 | |
7 | namespace torch { |
8 | namespace distributed { |
9 | namespace rpc { |
10 | |
11 | // Return value of a builtin operator or a TorchScript function. |
12 | class TORCH_API ScriptResp final : public RpcCommandBase { |
13 | public: |
14 | explicit ScriptResp(at::IValue&& values); |
15 | |
16 | const at::IValue& value(); |
17 | c10::intrusive_ptr<Message> toMessageImpl() && override; |
18 | static std::unique_ptr<ScriptResp> fromMessage(const Message& message); |
19 | |
20 | private: |
21 | const at::IValue value_; |
22 | }; |
23 | |
24 | } // namespace rpc |
25 | } // namespace distributed |
26 | } // namespace torch |
27 |