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

Method RemoveNode

tensorflow/compiler/jit/graphcycles/graphcycles.cc:129–140  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

127}
128
129void GraphCycles::RemoveNode(int32 node) {
130 Node* x = rep_->nodes_[node];
131 for (int32 y : x->out.GetSequence()) {
132 rep_->nodes_[y]->in.Erase(node);
133 }
134 for (int32 y : x->in.GetSequence()) {
135 rep_->nodes_[y]->out.Erase(node);
136 }
137 x->in.Clear();
138 x->out.Clear();
139 rep_->free_nodes_.push_back(node);
140}
141
142void* GraphCycles::GetNodeData(int32 node) const {
143 return rep_->nodes_[node]->data;

Callers 15

FunctionalizeLoopFunction · 0.45
TESTFunction · 0.45
ReplaceNodeFunction · 0.45
RearrangeRetvalNodesFunction · 0.45
MaybeRewriteWhileNodeFunction · 0.45
PartiallyDeclusterNodeFunction · 0.45

Calls 4

GetSequenceMethod · 0.80
EraseMethod · 0.45
ClearMethod · 0.45
push_backMethod · 0.45

Tested by 4

TESTFunction · 0.36
TESTFunction · 0.36
TESTFunction · 0.36
TEST_FFunction · 0.36