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

Function create_test_node

internal/bin/test_relu.cpp:87–116  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

Callers 1

create_test_graphFunction · 0.70

Calls 10

create_graph_nodeFunction · 0.85
get_graph_tensorFunction · 0.85
get_tensor_data_typeFunction · 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_floatFunction · 0.85
release_graph_nodeFunction · 0.85

Tested by

no test coverage detected