Searched refs:dst_device_ (Results 1 – 2 of 2) sorted by relevance
46 dst_device_(dst_device), in GdrRecvTensorCall()58 resp_.InitAlloc(dst_device_, recv_args_.alloc_attrs); in Start()65 const_cast<Tensor*>(&tensor()), transport_options, dst_device_, in Start()102 Device* dst_device() const { return dst_device_; } in dst_device()108 Device* dst_device_; member in tensorflow::__anon5c78cdd40111::GdrRecvTensorCall
59 RpcRecvTensorCall() : wi_(nullptr), dst_device_(nullptr) {} in RpcRecvTensorCall()66 dst_device_ = dst_device; in Init()81 dst_device_ = nullptr; in Reset()129 Device* dst_device() const { return dst_device_; } in dst_device()138 resp_.InitAlloc(dst_device_, alloc_attrs_); in StartRTCall()158 Device* dst_device_; member in tensorflow::__anon9a114d310111::RpcRecvTensorCall