| 37 | namespace cpu |
| 38 | { |
| 39 | void CpuActivation::configure(const ITensorInfo *input, ITensorInfo *output, const ActivationLayerInfo &activation_info) |
| 40 | { |
| 41 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuActivation::configure"); |
| 42 | ARM_COMPUTE_LOG_PARAMS(input, output, activation_info); |
| 43 | |
| 44 | _is_prepared = false; |
| 45 | auto k = std::make_unique<kernels::CpuActivationKernel>(); |
| 46 | k->configure(input, output, activation_info); |
| 47 | _kernel = std::move(k); |
| 48 | } |
| 49 | |
| 50 | Status |
| 51 | CpuActivation::validate(const ITensorInfo *input, const ITensorInfo *output, const ActivationLayerInfo &activation_info) |