| 5 | #include "tengine_c_api.h" |
| 6 | |
| 7 | int create_input_node(graph_t graph, const char* node_name, int c, int h, int w) |
| 8 | { |
| 9 | node_t node = create_graph_node(graph, node_name, "InputOp"); |
| 10 | tensor_t tensor = create_graph_tensor(graph, node_name, TENGINE_DT_FP32); |
| 11 | set_node_output_tensor(node, 0, tensor, TENSOR_TYPE_INPUT); |
| 12 | |
| 13 | int dims[4] = {1, c, h, w}; |
| 14 | |
| 15 | set_tensor_shape(tensor, dims, 4); |
| 16 | |
| 17 | release_graph_tensor(tensor); |
| 18 | release_graph_node(node); |
| 19 | |
| 20 | return 0; |
| 21 | } |
| 22 | |
| 23 | int create_conv_node(graph_t graph, const char* node_name, const char* input_name, int k_size, int stride, int pad, |
| 24 | int in_c, int out_c, int group) |
no test coverage detected