| 50 | } |
| 51 | |
| 52 | void schedule_kernel_on_ctx(IRuntimeContext *ctx, ICPPKernel *kernel, const IScheduler::Hints &hints) |
| 53 | { |
| 54 | if (ctx) |
| 55 | { |
| 56 | ARM_COMPUTE_ERROR_ON(ctx->scheduler() == nullptr); |
| 57 | ctx->scheduler()->schedule(kernel, hints); |
| 58 | } |
| 59 | else |
| 60 | { |
| 61 | NEScheduler::get().schedule(kernel, hints); |
| 62 | } |
| 63 | } |
| 64 | |
| 65 | unsigned int calculate_number_of_stages_only_x_axis(size_t input_x_dimension, unsigned int axis) |
| 66 | { |