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

Function AddControlTrigger

tensorflow/core/graph/graph_partition.cc:540–550  ·  view source on GitHub ↗

A dummy node for scheduling.

Source from the content-addressed store, hash-verified

538
539// A dummy node for scheduling.
540NodeDef* AddControlTrigger(const PartitionOptions& opts, GraphDef* gdef,
541 const string& assigned_device_name, int64 epoch,
542 int64 starttime, Status* status) {
543 NodeDef* result = gdef->add_node();
544 *status = NodeDefBuilder(opts.new_name(strings::StrCat("synch_", epoch)),
545 "ControlTrigger")
546 .Device(assigned_device_name)
547 .Attr("_start_time", starttime)
548 .Finalize(result, /*consume=*/true);
549 return result;
550}
551
552// Optimize colocation for control flow nodes. For cond, we want the
553// switch nodes to colocate with its data input. This is particularly

Callers 1

AddControlEdgesFunction · 0.85

Calls 6

NodeDefBuilderClass · 0.85
StrCatFunction · 0.50
FinalizeMethod · 0.45
AttrMethod · 0.45
DeviceMethod · 0.45
new_nameMethod · 0.45

Tested by

no test coverage detected