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

Method AddInputs

tensorflow/core/common_runtime/lower_if_op.cc:182–198  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

180}
181
182Status CondBuilder::AddInputs() {
183 // Add input data edges.
184 std::vector<const Edge*> edges;
185 TF_RETURN_IF_ERROR(if_op_->input_edges(&edges));
186 // Start at index 1 as the first input is the predicate.
187 for (int i = 1; i < edges.size(); ++i) {
188 const Edge* e = edges[i];
189 TF_RETURN_IF_ERROR(AddInput(e->src(), e->src_output()));
190 }
191 // Add input control edges.
192 for (const Edge* e : if_op_->in_edges()) {
193 if (e->IsControlEdge()) {
194 graph_->AddControlEdge(e->src(), control_predecessor_);
195 }
196 }
197 return Status::OK();
198}
199
200Status CondBuilder::AddOutputs() {
201 // Construct the then and else nodes.

Callers 1

RewriteIfNodeFunction · 0.45

Calls 7

input_edgesMethod · 0.80
AddInputFunction · 0.50
sizeMethod · 0.45
srcMethod · 0.45
src_outputMethod · 0.45
IsControlEdgeMethod · 0.45
AddControlEdgeMethod · 0.45

Tested by

no test coverage detected