@@ -17,9 +17,11 @@ std::uint32_t IEScheduler::num_threads() const {
1717 return parallel_get_max_threads ();
1818}
1919
20- void IEScheduler::Schedule (arm_compute::ICPPKernel* kernel, const arm_compute::IScheduler::Hints& hints, arm_compute::ITensorPack& tensors) {
20+ void IEScheduler::Schedule (arm_compute::ICPPKernel* kernel,
21+ const arm_compute::IScheduler::Hints& hints,
22+ const arm_compute::Window& max_window,
23+ arm_compute::ITensorPack& tensors) {
2124 IE_ASSERT (kernel != nullptr );
22- const auto & max_window = kernel->window ();
2325
2426 auto splitDimension = hints.split_dimension ();
2527
@@ -73,13 +75,14 @@ void IEScheduler::Schedule(arm_compute::ICPPKernel* kernel, const arm_compute::I
7375
7476void IEScheduler::schedule (arm_compute::ICPPKernel* kernel, const arm_compute::IScheduler::Hints& hints) {
7577 arm_compute::ITensorPack tensors;
76- Schedule (kernel, hints, tensors);
78+ Schedule (kernel, hints, kernel-> window (), tensors);
7779}
7880
7981void IEScheduler::schedule_op (arm_compute::ICPPKernel* kernel,
8082 const arm_compute::IScheduler::Hints& hints,
83+ const arm_compute::Window& window,
8184 arm_compute::ITensorPack& tensors) {
82- Schedule (kernel, hints, tensors);
85+ Schedule (kernel, hints, window, tensors);
8386}
8487
8588void IEScheduler::run_workloads (std::vector<arm_compute::IScheduler::Workload>& workloads) {
0 commit comments