| 55 | } |
| 56 | |
| 57 | void CpuActivation::run(ITensorPack &tensors) |
| 58 | { |
| 59 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuActivation::run"); |
| 60 | ARM_COMPUTE_ERROR_ON_MSG(tensors.empty(), "No inputs provided"); |
| 61 | |
| 62 | auto kernel_casted = static_cast<kernels::CpuActivationKernel *>(_kernel.get()); |
| 63 | if (!_is_prepared) |
| 64 | { |
| 65 | kernel_casted->prepare(tensors); |
| 66 | _is_prepared = true; |
| 67 | } |
| 68 | |
| 69 | const size_t split_dimension = kernel_casted->get_split_dimension_hint(); |
| 70 | NEScheduler::get().schedule_op(_kernel.get(), split_dimension, _kernel->window(), tensors); |
| 71 | } |
| 72 | |
| 73 | std::tuple<IOperator *, StatusCode> CpuContext::create_activation(const AclTensorDescriptor &src, |
| 74 | const AclTensorDescriptor &dst, |
nothing calls this directly
no test coverage detected