| 53 | NEConcatenateLayer::~NEConcatenateLayer() = default; |
| 54 | |
| 55 | void NEConcatenateLayer::configure(std::vector<const ITensor *> inputs_vector, ITensor *output, size_t axis) |
| 56 | { |
| 57 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "NEConcatenateLayer::configure"); |
| 58 | ARM_COMPUTE_ERROR_ON(output == nullptr); |
| 59 | |
| 60 | _impl->srcs = inputs_vector; |
| 61 | _impl->dst = output; |
| 62 | _impl->axis = axis; |
| 63 | _impl->num_inputs = inputs_vector.size(); |
| 64 | _impl->op = std::make_unique<cpu::CpuConcatenate>(); |
| 65 | |
| 66 | std::vector<const ITensorInfo *> inputs_vector_info; |
| 67 | for (unsigned int i = 0; i < inputs_vector.size(); ++i) |
| 68 | { |
| 69 | ARM_COMPUTE_ERROR_ON_NULLPTR(inputs_vector.at(i)); |
| 70 | inputs_vector_info.emplace_back(inputs_vector.at(i)->info()); |
| 71 | } |
| 72 | _impl->op->configure(inputs_vector_info, _impl->dst->info(), axis); |
| 73 | } |
| 74 | |
| 75 | Status NEConcatenateLayer::validate(const std::vector<const ITensorInfo *> &inputs_vector, |
| 76 | const ITensorInfo *output, |