Searched refs:dst_device_ (Results 1 – 1 of 1) sorted by relevance
60 RpcRecvTensorCall() : wi_(nullptr), dst_device_(nullptr) {} in RpcRecvTensorCall()67 dst_device_ = dst_device; in Init()82 dst_device_ = nullptr; in Reset()130 Device* dst_device() const { return dst_device_; } in dst_device()139 resp_.InitAlloc(dst_device_, alloc_attrs_); in StartRTCall()175 Device* dst_device_; member in tensorflow::__anona9090a8f0111::RpcRecvTensorCall