Searched refs:host_compute_sends_ (Results 1 – 2 of 2) sorted by relevance
1232 if (host_compute_sends_.find(key) != host_compute_sends_.end()) { in SetDeviceToHostMetadata()1236 tf2xla::HostTransferMetadata& transfer = host_compute_sends_[key]; in SetDeviceToHostMetadata()1243 const auto iter = host_compute_sends_.find(key); in GetDeviceToHostShapes()1244 if (iter == host_compute_sends_.end()) { in GetDeviceToHostShapes()1259 if (host_compute_recvs_.find(key) != host_compute_sends_.end()) { in SetHostToDeviceMetadata()
480 std::unordered_map<string, tf2xla::HostTransferMetadata> host_compute_sends_; variable