3 // This source code is licensed under the BSD-style license found in the
17 #include <xnnpack/node-type.h>
37     input_dims = RandomShape(4);  in DepthToSpaceTest()
40 …output_dims = {input_dims[0], input_dims[1] * block_size, input_dims[2] * block_size, output_chann…  in DepthToSpaceTest()
41     input_dims[3] = block_size * block_size * output_channels;  in DepthToSpaceTest()
44     input = std::vector<T>(NumElements(input_dims) + XNN_EXTRA_BYTES / sizeof(T));  in DepthToSpaceTest()
63     assert(input_dims[0] == output_dims[0]);  in batch_size()
64     return input_dims[0];  in batch_size()
67   size_t input_height() { return input_dims[1]; }  in input_height()
68   size_t input_width() { return input_dims[2]; }  in input_width()
69   size_t input_channel() { return input_dims[3]; }  in input_channel()
80   std::vector<size_t> input_dims;  member in DepthToSpaceTest
113 …                    subgraph, xnn_datatype_qint8, input_zero_point, input_scale, input_dims.size(),  in TEST_F()
114 …               input_dims.data(), nullptr, 0, /*flags=*/XNN_VALUE_FLAG_EXTERNAL_INPUT, &input_id));  in TEST_F()
126   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
127   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
128   ASSERT_EQ(node->type, xnn_node_type_depth_to_space);  in TEST_F()
129   ASSERT_EQ(node->compute_type, xnn_compute_type_qs8);  in TEST_F()
130   ASSERT_EQ(node->num_inputs, 1);  in TEST_F()
131   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
132   ASSERT_EQ(node->num_outputs, 1);  in TEST_F()
133   ASSERT_EQ(node->outputs[0], output_id);  in TEST_F()
134   ASSERT_EQ(node->flags, 0);  in TEST_F()
154 …                   subgraph, xnn_datatype_quint8, input_zero_point, input_scale, input_dims.size(),  in TEST_F()
155 …               input_dims.data(), nullptr, 0, /*flags=*/XNN_VALUE_FLAG_EXTERNAL_INPUT, &input_id));  in TEST_F()
167   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
168   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
169   ASSERT_EQ(node->type, xnn_node_type_depth_to_space);  in TEST_F()
170   ASSERT_EQ(node->compute_type, xnn_compute_type_qu8);  in TEST_F()
171   ASSERT_EQ(node->num_inputs, 1);  in TEST_F()
172   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
173   ASSERT_EQ(node->num_outputs, 1);  in TEST_F()
174   ASSERT_EQ(node->outputs[0], output_id);  in TEST_F()
175   ASSERT_EQ(node->flags, 0);  in TEST_F()
191 …                     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, 0,  in TEST_F()
204   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
205   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
206   ASSERT_EQ(node->type, xnn_node_type_depth_to_space);  in TEST_F()
207   ASSERT_EQ(node->compute_type, xnn_compute_type_fp32);  in TEST_F()
208   ASSERT_EQ(node->num_inputs, 1);  in TEST_F()
209   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
210   ASSERT_EQ(node->num_outputs, 1);  in TEST_F()
211   ASSERT_EQ(node->outputs[0], output_id);  in TEST_F()
212   ASSERT_EQ(node->flags, 0);  in TEST_F()
252 …subgraph, xnn_datatype_qint8, input_zero_point, input_scale, input_dims.size(), input_dims.data(),…  in TEST_F()
316 …subgraph, xnn_datatype_quint8, input_zero_point, input_scale, input_dims.size(), input_dims.data()…  in TEST_F()
345   std::uniform_real_distribution<float> f32dist(-255.0f, 255.0f);  in TEST_F()
378 …     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, /*external_id=*/0,  in TEST_F()