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

Method CreateMergeNodes

tensorflow/core/common_runtime/lower_while_op.cc:240–252  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

238}
239
240Status LowerWhileHelper::CreateMergeNodes() {
241 for (Node* enter_node : enter_nodes_) {
242 Node* merge_node;
243 TF_RETURN_IF_ERROR(
244 NodeBuilder(NewName("merge"), "Merge", graph_->op_registry(),
245 &debug_info_)
246 .Input({NodeOut(enter_node, 0), NodeOut(enter_node, 0)})
247 .Device(while_op_->requested_device())
248 .Finalize(graph_, &merge_node));
249 merge_nodes_.emplace_back(merge_node);
250 }
251 return Status::OK();
252}
253
254Status LowerWhileHelper::CreateCondFuncCallNode() {
255 for (Node* merge_node : merge_nodes_) {

Callers

nothing calls this directly

Calls 8

NewNameFunction · 0.85
NodeBuilderClass · 0.50
NodeOutClass · 0.50
FinalizeMethod · 0.45
DeviceMethod · 0.45
InputMethod · 0.45
op_registryMethod · 0.45
emplace_backMethod · 0.45

Tested by

no test coverage detected