MCPcopy Create free account
hub / github.com/OAID/Tengine / create_input_node

Function create_input_node

core/lib/auth_helper.cpp:39–53  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

37namespace AuthHelp
38{
39 int create_input_node(graph_t graph, const char* node_name, int c, int h, int w)
40 {
41 node_t node = create_graph_node(graph, node_name, "InputOp");
42 tensor_t tensor = create_graph_tensor(graph, node_name, TENGINE_DT_FP32);
43 set_node_output_tensor(node, 0, tensor, TENSOR_TYPE_INPUT);
44
45 int dims[4] = {1, c, h, w};
46
47 set_tensor_shape(tensor, dims, 4);
48
49 release_graph_tensor(tensor);
50 release_graph_node(node);
51
52 return 0;
53 }
54
55
56 int create_pool_node(graph_t graph, const char* node_name, const char* input_name, int kernel_h, int kernel_w,

Callers 1

create_pool_graphFunction · 0.70

Calls 6

create_graph_nodeFunction · 0.85
create_graph_tensorFunction · 0.85
set_node_output_tensorFunction · 0.85
set_tensor_shapeFunction · 0.85
release_graph_tensorFunction · 0.85
release_graph_nodeFunction · 0.85

Tested by

no test coverage detected