Skip to content

T
tic
Commit 97811fa4 by Eric Junyuan Xie Committed by Tianqi Chen
Options

fix (#138)

parent 51f71ea1
Showing with 1 additions and 0 deletions
nnvm/include/nnvm/graph.h
......@@ -285,6 +285,7 @@ inline void DFSVisit(const std::vector<NodeEntry>& heads,
[fvisit](GNode n) { fvisit(*n); }, // FVisit
[](GNode n)->Node* { return n->get(); }, // HashFunc
[](GNode n)->uint32_t { // InDegree
if (!(*n)) return 0;
return (*n)->inputs.size() + (*n)->control_deps.size();
},
[](GNode n, uint32_t index)->GNode { // GetInput
......
Markdown is supported
0% or
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment