| 2202 | } |
| 2203 | |
| 2204 | void compute_maxpool1d_node(GraphNode& node, const std::vector<std::unique_ptr<GraphNode>>& nodes, |
| 2205 | const std::unordered_map<size_t, size_t>& node_index_map) { |
| 2206 | const auto& input = get_input(node, 0, nodes, node_index_map); |
| 2207 | |
| 2208 | size_t batch_size = input.shape[0]; |
| 2209 | size_t channels = input.shape[1]; |
| 2210 | size_t input_length = input.shape[2]; |
| 2211 | size_t kernel_size = node.params.kernel_size; |
| 2212 | size_t stride = node.params.stride; |
| 2213 | |
| 2214 | cactus_maxpool1d_f16( |
| 2215 | input.data_as<__fp16>(), |
| 2216 | node.output_buffer.data_as<__fp16>(), |
| 2217 | batch_size, channels, input_length, |
| 2218 | kernel_size, stride); |
| 2219 | } |
| 2220 | |
| 2221 | void compute_conv2d_k3s1p1_node(GraphNode& node, const std::vector<std::unique_ptr<GraphNode>>& nodes, |
| 2222 | const std::unordered_map<size_t, size_t>& node_index_map) { |
nothing calls this directly
no test coverage detected