3 // This source code is licensed under the BSD-style license found in the
16 #include <xnnpack/node-type.h>
40     input_dims = output1_dims;  in EvenSplit2Test()
41     input_dims[axis] = output1_dims[axis] + output2_dims[axis];  in EvenSplit2Test()
43     input = std::vector<T>(NumElements(input_dims));  in EvenSplit2Test()
56       batch_size *= input_dims[i];  in EvenSplit2Test()
59     for (size_t i = axis; i < input_dims.size(); i++) {  in EvenSplit2Test()
60       input_stride *= input_dims[i];  in EvenSplit2Test()
74     return std::uniform_int_distribution<size_t>(0, dims.size() - 1)(rng);  in RandomAxis()
97   std::vector<size_t> input_dims;  member in EvenSplit2Test
131 …subgraph, xnn_datatype_qint8, signed_zero_point, scale, input_dims.size(), input_dims.data(), null…  in TEST_F()
153   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
154   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
155   ASSERT_EQ(node->type, xnn_node_type_even_split2);  in TEST_F()
156   ASSERT_EQ(node->compute_type, xnn_compute_type_qs8);  in TEST_F()
157   ASSERT_EQ(node->params.even_split.axis, axis);  in TEST_F()
158   ASSERT_EQ(node->num_inputs, 1);  in TEST_F()
159   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
160   ASSERT_EQ(node->num_outputs, 2);  in TEST_F()
161   ASSERT_EQ(node->outputs[0], output1_id);  in TEST_F()
162   ASSERT_EQ(node->outputs[1], output2_id);  in TEST_F()
163   ASSERT_EQ(node->flags, 0);  in TEST_F()
178 …subgraph, xnn_datatype_quint8, unsigned_zero_point, scale, input_dims.size(), input_dims.data(), n…  in TEST_F()
200   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
201   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
202   ASSERT_EQ(node->type, xnn_node_type_even_split2);  in TEST_F()
203   ASSERT_EQ(node->compute_type, xnn_compute_type_qu8);  in TEST_F()
204   ASSERT_EQ(node->params.even_split.axis, axis);  in TEST_F()
205   ASSERT_EQ(node->num_inputs, 1);  in TEST_F()
206   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
207   ASSERT_EQ(node->num_outputs, 2);  in TEST_F()
208   ASSERT_EQ(node->outputs[0], output1_id);  in TEST_F()
209   ASSERT_EQ(node->outputs[1], output2_id);  in TEST_F()
210   ASSERT_EQ(node->flags, 0);  in TEST_F()
224 …                     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, 0,  in TEST_F()
243   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
244   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
245   ASSERT_EQ(node->type, xnn_node_type_even_split2);  in TEST_F()
246   ASSERT_EQ(node->compute_type, xnn_compute_type_fp32);  in TEST_F()
247   ASSERT_EQ(node->params.even_split.axis, axis);  in TEST_F()
248   ASSERT_EQ(node->num_inputs, 1);  in TEST_F()
249   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
250   ASSERT_EQ(node->num_outputs, 2);  in TEST_F()
251   ASSERT_EQ(node->outputs[0], output1_id);  in TEST_F()
252   ASSERT_EQ(node->outputs[1], output2_id);  in TEST_F()
253   ASSERT_EQ(node->flags, 0);  in TEST_F()
281 …op2, batch_size, (uint8_t*) input.data() + op1->channels, operator_output2.data(), nullptr /* thre…  in TEST_F()
295 …subgraph, xnn_datatype_qint8, signed_zero_point, scale, input_dims.size(), input_dims.data(), null…  in TEST_F()
358 …op2, batch_size, (uint8_t*) input.data() + op1->channels, operator_output2.data(), nullptr /* thre…  in TEST_F()
372 …subgraph, xnn_datatype_quint8, unsigned_zero_point, scale, input_dims.size(), input_dims.data(), n…  in TEST_F()
435 …op2, batch_size, (uint32_t*) input.data() + op1->channels, operator_output2.data(), nullptr /* thr…  in TEST_F()
448 …                     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, 0,  in TEST_F()