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

Function create_test_node

internal/bin/test_resize_op.cpp:54–107  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

52 return 0;
53}
54int create_test_node(graph_t graph, const char* node_name, const char* input_name)
55{
56 node_t test_node = create_graph_node(graph, node_name, "Resize");
57
58 const int type = 1;
59 set_node_attr_int(test_node, "type", &type);
60 const float scale_w = 0.4;
61 const float scale_h = 0.4;
62 set_node_attr_float(test_node, "scale_w", &scale_w);
63 set_node_attr_float(test_node, "scale_h", &scale_h);
64
65 tensor_t input_tensor = get_graph_tensor(graph, input_name);
66 int data_type = get_tensor_data_type(input_tensor);
67
68 if(input_tensor == nullptr)
69 {
70 std::cout << "ERRNO: " << get_tengine_errno() << "\n";
71 return -1;
72 }
73
74 set_node_input_tensor(test_node, 0, input_tensor);
75
76 release_graph_tensor(input_tensor);
77
78 /* output */
79 tensor_t output_tensor = create_graph_tensor(graph, node_name, data_type);
80
81 set_node_output_tensor(test_node, 0, output_tensor, TENSOR_TYPE_VAR);
82
83 release_graph_tensor(output_tensor);
84
85 // // std::string slope_name(node_name);
86 // // slope_name += "/slope";
87
88 // node_t s_node = create_graph_node(graph, "slope", "Const");
89 // tensor_t s_tensor = create_graph_tensor(graph, "slope", TENGINE_DT_FP32);
90 // set_node_output_tensor(s_node, 0, s_tensor, TENSOR_TYPE_CONST);
91
92 // set_node_input_tensor(test_node, 1, s_tensor);
93 // int dims[4];
94 // get_tensor_shape(input_tensor,dims,4);
95 // int dim1=dims[1];
96 // int s_dims[] = {dim1};
97
98 // set_tensor_shape(s_tensor, s_dims, 1);
99
100 // release_graph_node(s_node);
101 // release_graph_tensor(s_tensor);
102
103 // set_node_attr_float(test_node, "negative_slope", &negative_slope);
104 release_graph_node(test_node);
105
106 return 0;
107}
108
109graph_t create_test_graph(const char* test_node_name, int c, int h, int w, int layout, int data_type)
110{

Callers 1

create_test_graphFunction · 0.70

Calls 11

create_graph_nodeFunction · 0.85
set_node_attr_intFunction · 0.85
set_node_attr_floatFunction · 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
release_graph_nodeFunction · 0.85

Tested by

no test coverage detected