diff options
Diffstat (limited to 'src/cpu/kernels/CpuReshapeKernel.h')
-rw-r--r-- | src/cpu/kernels/CpuReshapeKernel.h | 7 |
1 files changed, 3 insertions, 4 deletions
diff --git a/src/cpu/kernels/CpuReshapeKernel.h b/src/cpu/kernels/CpuReshapeKernel.h index eddbbf7135..ce566fd9e2 100644 --- a/src/cpu/kernels/CpuReshapeKernel.h +++ b/src/cpu/kernels/CpuReshapeKernel.h @@ -55,7 +55,7 @@ public: static Status validate(const ITensorInfo *src, const ITensorInfo *dst); // Inherited methods overridden: - void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override; + void run_op(ITensorPack &tensors, const Window &window, const ThreadInfo &info) override; const char *name() const override; /** Prepare the reshape kernel for execution (Only executed once) by calculating max or squashed window and selecting the _reshape_tensor_fn based on the presence of holes @@ -84,10 +84,9 @@ public: } private: - size_t _split_dimension{ Window::DimY }; - - std::function<void(const Window &window, const ITensor *src, ITensor *dst )> _reshape_tensor_fn{}; + size_t _split_dimension{Window::DimY}; + std::function<void(const Window &window, const ITensor *src, ITensor *dst)> _reshape_tensor_fn{}; }; } // namespace kernels } // namespace cpu |