(cur, id2node, name2id, ops, model)
| 143 | |
| 144 | |
| 145 | def Prune2Sparse(cur, id2node, name2id, ops, model): |
| 146 | # Assume that FC and Relu takes in only 1 input; |
| 147 | # If not raise warning |
| 148 | if not cur.visited and cur.optype == "FC_Prune": |
| 149 | transFCRelu(cur, id2node, name2id, ops, model) |
| 150 | |
| 151 | cur.visited = True |
| 152 | for name, n in cur.ops.iteritems(): |
| 153 | Prune2Sparse(n, id2node, name2id, ops, model) |
| 154 | |
| 155 | |
| 156 | def net2list(net_root): |
nothing calls this directly
no test coverage detected
searching dependent graphs…