| 14 | namespace embedx { |
| 15 | |
| 16 | GraphNode* GetXInput(const std::string& name) { |
| 17 | // Shape (BATCH_PLACEHOLDER = 100, ) cannot reshape to (n,3) when compiling |
| 18 | // the graph. Change X shape from (BATCH_PLACEHOLDER, ) to (-1, ). |
| 19 | return new InstanceNode(name, Shape(-1, 0), TENSOR_TYPE_CSR); |
| 20 | } |
| 21 | |
| 22 | std::vector<GraphNode*> GetXBlockInputs(const std::string& name, int depth) { |
| 23 | std::vector<GraphNode*> inst_nodes; |
no outgoing calls
no test coverage detected