MCPcopy Create free account
hub / github.com/DeepRec-AI/DeepRec / GetChannelHandle

Method GetChannelHandle

tensorflow/compiler/tf2xla/xla_compiler.cc:1313–1322  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

1311}
1312
1313Status XlaCompiler::GetChannelHandle(const string& key,
1314 xla::ChannelHandle* channel) {
1315 auto result = channels_.emplace(key, xla::ChannelHandle());
1316 if (result.second) {
1317 TF_ASSIGN_OR_RETURN(result.first->second, client()->CreateChannelHandle());
1318 }
1319 *channel = result.first->second;
1320 VLOG(1) << "Channel: " << key << " " << channel->DebugString();
1321 return Status::OK();
1322}
1323
1324Status XlaCompiler::GetHostToDeviceChannelHandle(const string& key,
1325 xla::ChannelHandle* channel) {

Callers 1

CompileMethod · 0.80

Calls 4

TF_ASSIGN_OR_RETURNFunction · 0.50
emplaceMethod · 0.45
CreateChannelHandleMethod · 0.45
DebugStringMethod · 0.45

Tested by

no test coverage detected