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

Function AddReadControl

tensorflow/core/graph/graph_partition.cc:160–167  ·  view source on GitHub ↗

Add a control edge from each input to each recv.

Source from the content-addressed store, hash-verified

158
159// Add a control edge from each input to each recv.
160void AddReadControl(const std::vector<NodeDef*>& recvs,
161 const std::vector<string>& inputs) {
162 for (NodeDef* recv : recvs) {
163 for (const string& input : inputs) {
164 recv->add_input(strings::StrCat("^", input));
165 }
166 }
167}
168
169void SetSendRecvAttrs(const PartitionOptions& opts, const Edge* edge,
170 NodeDefBuilder* builder) {

Callers 2

PartitionFunction · 0.85
PartitionWithTensorFuseFunction · 0.85

Calls 2

StrCatFunction · 0.50
add_inputMethod · 0.45

Tested by

no test coverage detected