| 1322 | } |
| 1323 | |
| 1324 | Status XlaCompiler::GetHostToDeviceChannelHandle(const string& key, |
| 1325 | xla::ChannelHandle* channel) { |
| 1326 | auto result = channels_.emplace(key, xla::ChannelHandle()); |
| 1327 | if (result.second) { |
| 1328 | TF_ASSIGN_OR_RETURN(result.first->second, |
| 1329 | client()->CreateHostToDeviceChannelHandle()); |
| 1330 | } |
| 1331 | *channel = result.first->second; |
| 1332 | VLOG(1) << "Host to device channel: " << key << " " << channel->DebugString(); |
| 1333 | return Status::OK(); |
| 1334 | } |
| 1335 | |
| 1336 | Status XlaCompiler::GetDeviceToHostChannelHandle(const string& key, |
| 1337 | xla::ChannelHandle* channel) { |
nothing calls this directly
no test coverage detected