Searched refs:host_compute_sends_ (Results 1 – 2 of 2) sorted by relevance
353 std::unordered_map<string, tf2xla::HostTransferMetadata> host_compute_sends_; variable
1458 if (host_compute_sends_.find(key) != host_compute_sends_.end()) { in SetDeviceToHostMetadata()1459 tf2xla::HostTransferMetadata& existing_transfer = host_compute_sends_[key]; in SetDeviceToHostMetadata()1469 tf2xla::HostTransferMetadata& transfer = host_compute_sends_[key]; in SetDeviceToHostMetadata()1476 const auto iter = host_compute_sends_.find(key); in GetDeviceToHostShapes()1477 if (iter == host_compute_sends_.end()) { in GetDeviceToHostShapes()