| 49 | } |
| 50 | |
| 51 | void CpuReshape::run(ITensorPack &tensors) |
| 52 | { |
| 53 | ARM_COMPUTE_TRACE_EVENT(ARM_COMPUTE_PROF_CAT_CPU, ARM_COMPUTE_PROF_LVL_CPU, "CpuReshape::run"); |
| 54 | ARM_COMPUTE_ERROR_ON_MSG(tensors.empty(), "No inputs provided"); |
| 55 | if (!_is_prepared) |
| 56 | { |
| 57 | static_cast<kernels::CpuReshapeKernel *>(_kernel.get())->prepare(tensors); |
| 58 | _is_prepared = true; |
| 59 | } |
| 60 | const auto split_dimension = static_cast<kernels::CpuReshapeKernel *>(_kernel.get())->get_split_dimension(); |
| 61 | NEScheduler::get().schedule_op(_kernel.get(), split_dimension, _kernel->window(), tensors); |
| 62 | } |
| 63 | } // namespace cpu |
| 64 | } // namespace arm_compute |
nothing calls this directly
no test coverage detected