| 45 | } |
| 46 | |
| 47 | void NEStackLayer::configure(const std::vector<ITensor *> &input, int axis, ITensor *output) |
| 48 | { |
| 49 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "NEStackLayer::configure"); |
| 50 | ARM_COMPUTE_LOG_PARAMS(input, axis, output); |
| 51 | |
| 52 | // Wrap around negative values |
| 53 | const unsigned int axis_u = wrap_around(axis, static_cast<int>(input[0]->info()->num_dimensions() + 1)); |
| 54 | |
| 55 | _stack_kernel->configure(input, axis_u, output); |
| 56 | } |
| 57 | |
| 58 | Status NEStackLayer::validate(const std::vector<ITensorInfo *> &input, int axis, const ITensorInfo *output) |
| 59 | { |
nothing calls this directly
no test coverage detected