38 struct NEConcatenateLayer::Impl
40 std::vector<const ITensor *> srcs{};
41 ITensor *
dst{
nullptr};
42 unsigned int num_inputs{0};
44 std::unique_ptr<cpu::CpuConcatenate> op{
nullptr};
58 _impl->srcs = inputs_vector;
61 _impl->num_inputs = inputs_vector.size();
62 _impl->op = std::make_unique<cpu::CpuConcatenate>();
64 std::vector<const ITensorInfo *> inputs_vector_info;
65 for (
unsigned int i = 0; i < inputs_vector.size(); ++i)
68 inputs_vector_info.emplace_back(inputs_vector.at(i)->info());
70 _impl->op->configure(inputs_vector_info, _impl->dst->info(), axis);
83 for (
unsigned i = 0; i < _impl->num_inputs; ++i)