Home
last modified time | relevance | path

Searched refs:host_compute_recvs_ (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler.h354 std::unordered_map<string, tf2xla::HostTransferMetadata> host_compute_recvs_; variable
Dxla_compiler.cc1492 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()