| 1389 | } |
| 1390 | |
| 1391 | Status XlaCompiler::SetHostToDeviceMetadata( |
| 1392 | const string& key, absl::Span<const DataType> types, |
| 1393 | absl::Span<const TensorShape> shapes) { |
| 1394 | if (host_compute_recvs_.find(key) != host_compute_sends_.end()) { |
| 1395 | return errors::InvalidArgument( |
| 1396 | "Duplicate calls to SetHostToDeviceMetadata with key ", key); |
| 1397 | } |
| 1398 | tf2xla::HostTransferMetadata& transfer = host_compute_recvs_[key]; |
| 1399 | SetTransfer(key, types, shapes, &transfer); |
| 1400 | return Status::OK(); |
| 1401 | } |
| 1402 | |
| 1403 | Status XlaCompiler::GetHostComputeControlDependency( |
| 1404 | const string& host_compute_name, xla::XlaOp* handle) { |
nothing calls this directly
no test coverage detected