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