| 1362 | } // namespace |
| 1363 | |
| 1364 | Status XlaCompiler::SetDeviceToHostMetadata( |
| 1365 | const string& key, absl::Span<const DataType> types, |
| 1366 | absl::Span<const TensorShape> shapes) { |
| 1367 | if (host_compute_sends_.find(key) != host_compute_sends_.end()) { |
| 1368 | return errors::InvalidArgument( |
| 1369 | "Duplicate calls to SetDeviceToHostMetadata with key ", key); |
| 1370 | } |
| 1371 | tf2xla::HostTransferMetadata& transfer = host_compute_sends_[key]; |
| 1372 | SetTransfer(key, types, shapes, &transfer); |
| 1373 | return Status::OK(); |
| 1374 | } |
| 1375 | |
| 1376 | Status XlaCompiler::GetDeviceToHostShapes( |
| 1377 | const string& key, std::vector<TensorShape>* shapes) const { |
nothing calls this directly
no test coverage detected