1 #pragma once 2 3 #include <torch/csrc/distributed/rpc/message.h> 4 #include <torch/csrc/distributed/rpc/rpc_command_base.h> 5 6 namespace torch { 7 namespace distributed { 8 namespace autograd { 9 10 // Response for the RRefBackwardReq. 11 class TORCH_API RRefBackwardResp : public rpc::RpcCommandBase { 12 public: 13 RRefBackwardResp() = default; 14 c10::intrusive_ptr<rpc::Message> toMessageImpl() && override; 15 static std::unique_ptr<RRefBackwardResp> fromMessage( 16 const rpc::Message& message); 17 }; 18 19 } // namespace autograd 20 } // namespace distributed 21 } // namespace torch 22