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

Function create_test_node

internal/bin/test_interp.cpp:86–121  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

84 return 0;
85}
86int create_test_node(graph_t graph, const char* node_name, const char* input_name, int data_type, int resize_type)
87{
88 node_t test_node = create_graph_node(graph, node_name, "Interp");
89
90 tensor_t input_tensor = get_graph_tensor(graph, input_name);
91
92 if(input_tensor == nullptr)
93 {
94 std::cout << "ERRNO: " << get_tengine_errno() << "\n";
95 return -1;
96 }
97
98 set_node_input_tensor(test_node, 0, input_tensor);
99
100 release_graph_tensor(input_tensor);
101
102 /* output */
103 tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);
104
105 set_node_output_tensor(test_node, 0, output_tensor, TENSOR_TYPE_VAR);
106
107 release_graph_tensor(output_tensor);
108
109 int output_width = 0;
110 int output_height = 0;
111 float width_scale = 2.0;
112 float height_scale = 2.0;
113 set_node_attr_int(test_node, "resize_type", &resize_type);
114 set_node_attr_int(test_node, "output_width", &output_width);
115 set_node_attr_int(test_node, "output_height", &output_height);
116 set_node_attr_float(test_node, "width_scale", &width_scale);
117 set_node_attr_float(test_node, "height_scale", &height_scale);
118 release_graph_node(test_node);
119
120 return 0;
121}
122
123graph_t create_test_graph(const char* test_node_name, int c, int h, int w, int layout, int data_type, int resize_type)
124{

Callers 1

create_test_graphFunction · 0.70

Calls 10

create_graph_nodeFunction · 0.85
get_graph_tensorFunction · 0.85
get_tengine_errnoFunction · 0.85
set_node_input_tensorFunction · 0.85
release_graph_tensorFunction · 0.85
create_graph_tensorFunction · 0.85
set_node_output_tensorFunction · 0.85
set_node_attr_intFunction · 0.85
set_node_attr_floatFunction · 0.85
release_graph_nodeFunction · 0.85

Tested by

no test coverage detected