3 // This source code is licensed under the BSD-style license found in the
16 #include <xnnpack/node-type.h>
29     input_dims = RandomShape(4);  in SetUp()
30     output_dims = input_dims;  in SetUp()
31     batch_size = input_dims[0] * input_dims[1] * input_dims[2];  in SetUp()
32     channels = input_dims[3];  in SetUp()
34     input = std::vector<float>(XNN_EXTRA_BYTES / sizeof(float) + NumElements(input_dims));  in SetUp()
57   std::vector<size_t> input_dims;  member in PreluTestF32
78 …                     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, 0,  in TEST_F()
92 …                     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, 2,  in TEST_F()
98   ASSERT_EQ(subgraph->num_nodes, 1);  in TEST_F()
99   const struct xnn_node* node = &subgraph->nodes[0];  in TEST_F()
100   ASSERT_EQ(node->type, xnn_node_type_prelu);  in TEST_F()
101   ASSERT_EQ(node->compute_type, xnn_compute_type_fp32);  in TEST_F()
102   ASSERT_EQ(node->num_inputs, 2);  in TEST_F()
103   ASSERT_EQ(node->inputs[0], input_id);  in TEST_F()
104   ASSERT_EQ(node->inputs[1], slope_id);  in TEST_F()
105   ASSERT_EQ(node->num_outputs, 1);  in TEST_F()
106   ASSERT_EQ(node->outputs[0], output_id);  in TEST_F()
107   ASSERT_EQ(node->flags, 0);  in TEST_F()
112   std::uniform_real_distribution<float> f32idist(-1.0f, 1.0f);  in TEST_F()
146 …     subgraph, xnn_datatype_fp32, input_dims.size(), input_dims.data(), nullptr, /*external_id=*/0,  in TEST_F()