Searched defs:wrappedMessageType (Results 1 – 4 of 4) sorted by relevance
39 MessageType wrappedMessageType, in RpcWithAutograd()57 auto wrappedMessageType = wrappedMessage_->type(); in toMessageImpl() local107 MessageType wrappedMessageType = in fromMessage() local165 MessageType RpcWithAutograd::wrappedMessageType() const { in wrappedMessageType() function in torch::distributed::autograd::RpcWithAutograd
40 rpc::MessageType wrappedMessageType, in RpcWithProfilingReq()53 rpc::MessageType RpcWithProfilingReq::wrappedMessageType() const { in wrappedMessageType() function in torch::distributed::autograd::RpcWithProfilingReq
34 rpc::MessageType wrappedMessageType, in RpcWithProfilingResp()52 rpc::MessageType RpcWithProfilingResp::wrappedMessageType() const { in wrappedMessageType() function in torch::distributed::autograd::RpcWithProfilingResp
313 auto wrappedMessageType = rpcWithAutograd.wrappedMessageType(); in processForwardAutogradReq() local