diff options
Diffstat (limited to 'src/cpu/kernels/CpuSubKernel.cpp')
-rw-r--r-- | src/cpu/kernels/CpuSubKernel.cpp | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuSubKernel.cpp b/src/cpu/kernels/CpuSubKernel.cpp index c55d11e899..d908e4ed28 100644 --- a/src/cpu/kernels/CpuSubKernel.cpp +++ b/src/cpu/kernels/CpuSubKernel.cpp @@ -131,7 +131,8 @@ void CpuSubKernel::configure(const ITensorInfo *src0, const ITensorInfo *src1, I _name = std::string("CpuSubKernel").append("/").append(uk->name); // CpuSubKernel doesn't need padding so update_window_and_padding() can be skipped - Window win = calculate_max_window(out_shape, Steps()); + Window win; + std::tie(win, _split_dimension) = calculate_squashed_or_max_window(*src0, *src1); ICpuKernel::configure(win); } |