Note: Only returns 1 output
| 156 | |
| 157 | // Note: Only returns 1 output |
| 158 | py::dict outputPyArray(const luci::CircleNode *node, luci_interpreter::Interpreter *interpreter) |
| 159 | { |
| 160 | assert(node != nullptr); // FIX_CALLER_UNLESS |
| 161 | assert(interpreter != nullptr); // FIX_CALLER_UNLESS |
| 162 | |
| 163 | const auto tensor = interpreter->getTensor(node); |
| 164 | |
| 165 | THROW_UNLESS(tensor != nullptr, "Null tensor detected in " + node->name()); |
| 166 | |
| 167 | auto py_output = py::dict("name"_a = node->name(), "data"_a = numpyArray(tensor), |
| 168 | "quantparam"_a = quantparam(tensor), |
| 169 | "is_const"_a = node->opcode() == luci::CircleOpcode::CIRCLECONST); |
| 170 | return py_output; |
| 171 | } |
| 172 | |
| 173 | bool multi_out_node(const luci::CircleNode *node) |
| 174 | { |
no test coverage detected