Searched refs:host_compute_recvs_ (Results 1 – 2 of 2) sorted by relevance
354 std::unordered_map<string, tf2xla::HostTransferMetadata> host_compute_recvs_; variable
1492 if (host_compute_recvs_.find(key) != host_compute_recvs_.end()) { in SetHostToDeviceMetadata()1493 tf2xla::HostTransferMetadata& existing_transfer = host_compute_recvs_[key]; in SetHostToDeviceMetadata()1503 tf2xla::HostTransferMetadata& transfer = host_compute_recvs_[key]; in SetHostToDeviceMetadata()