| 222 | } |
| 223 | |
| 224 | void ClConcatenate::run(ITensorPack &tensors) |
| 225 | { |
| 226 | if (tensors.empty()) |
| 227 | { |
| 228 | ARM_COMPUTE_ERROR("No inputs provided"); |
| 229 | } |
| 230 | |
| 231 | if (static_cast<int>(tensors.size()) - 1 != static_cast<int>(_num_inputs)) |
| 232 | { |
| 233 | ARM_COMPUTE_ERROR("Configured with different number of inputs"); |
| 234 | } |
| 235 | |
| 236 | if (_axis == Window::DimX && (_num_inputs == 2 || _num_inputs == 4)) |
| 237 | { |
| 238 | ARM_COMPUTE_ERROR_ON(_concat_kernels.empty()); |
| 239 | CLScheduler::get().enqueue_op(*_concat_kernels.at(0), tensors, true); |
| 240 | } |
| 241 | else |
| 242 | { |
| 243 | int i = 0; |
| 244 | for (auto &k : _concat_kernels) |
| 245 | { |
| 246 | ITensorPack pack; |
| 247 | pack.add_tensor(TensorType::ACL_SRC, tensors.get_const_tensor(ACL_SRC_VEC + i)); |
| 248 | pack.add_tensor(TensorType::ACL_DST, tensors.get_tensor(ACL_DST)); |
| 249 | CLScheduler::get().enqueue_op(*k, pack, true); |
| 250 | ++i; |
| 251 | } |
| 252 | } |
| 253 | } |
| 254 | } // namespace opencl |
| 255 | } // namespace arm_compute |
nothing calls this directly
no test coverage detected