| 92 | } |
| 93 | |
| 94 | std::string dnnc::cppCodeGen::nodeName(node *n) { |
| 95 | if (n->ntype() == node::OPERATOR) |
| 96 | return _prefix + cppName(n->name()) + "_" + |
| 97 | static_cast<opNode *>(n)->outputs()[0]; |
| 98 | else if (n->ntype() == node::INPUT) |
| 99 | return _prefix + cppName(n->name()); |
| 100 | else if (n->ntype() == node::OUTPUT) |
| 101 | return _prefix + cppName(n->name()); |
| 102 | else |
| 103 | assert(false); |
| 104 | return _prefix + cppName(n->name()); |
| 105 | } |
| 106 | |
| 107 | std::string dnnc::cppCodeGen::shapeStr(std::vector<DIMENSION> shapeVec) { |
| 108 | std::string shapeStr; |