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

Function create_test_node

internal/bin/test_prelu.cpp:86–131  ·  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)
87{
88 node_t test_node = create_graph_node(graph, node_name, "PReLU");
89
90 tensor_t input_tensor = get_graph_tensor(graph, input_name);
91 int data_type = get_tensor_data_type(input_tensor);
92
93 if(input_tensor == nullptr)
94 {
95 std::cout << "ERRNO: " << get_tengine_errno() << "\n";
96 return -1;
97 }
98
99 set_node_input_tensor(test_node, 0, input_tensor);
100
101 release_graph_tensor(input_tensor);
102
103 /* output */
104 tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);
105
106 set_node_output_tensor(test_node, 0, output_tensor, TENSOR_TYPE_VAR);
107
108 release_graph_tensor(output_tensor);
109
110 // std::string slope_name(node_name);
111 // slope_name += "/slope";
112
113 node_t s_node = create_graph_node(graph, "slope", "Const");
114 tensor_t s_tensor = create_graph_tensor(graph, "slope", TENGINE_DT_FP32);
115 set_node_output_tensor(s_node, 0, s_tensor, TENSOR_TYPE_CONST);
116
117 set_node_input_tensor(test_node, 1, s_tensor);
118 int dims[4];
119 get_tensor_shape(input_tensor, dims, 4);
120 int dim1 = dims[1];
121 int s_dims[1] = {dim1};
122
123 set_tensor_shape(s_tensor, s_dims, 1);
124
125 release_graph_node(s_node);
126 release_graph_tensor(s_tensor);
127
128 release_graph_node(test_node);
129
130 return 0;
131}
132
133graph_t create_test_graph(const char* test_node_name, int c, int h, int w, int layout, int data_type)
134{

Callers 1

create_test_graphFunction · 0.70

Calls 11

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
get_tensor_shapeFunction · 0.85
set_tensor_shapeFunction · 0.85
release_graph_nodeFunction · 0.85

Tested by

no test coverage detected