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

Function create_test_node

internal/bin/test_elu.cpp:85–112  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

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

Callers 1

create_test_graphFunction · 0.70

Calls 9

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

Tested by

no test coverage detected