Update reinterpret tensor as 1D for CPU add
* Use the same implementation as other layers.
Resolves: COMPMID-5108
Signed-off-by: Viet-Hoa Do <viet-hoa.do@arm.com>
Change-Id: I5a50259b398b71ca1f61b5ee8daa539bf8263fac
Reviewed-on: https://review.mlplatform.org/c/ml/ComputeLibrary/+/8501
Benchmark: Arm Jenkins <bsgcomp@arm.com>
Tested-by: Arm Jenkins <bsgcomp@arm.com>
Reviewed-by: Gunes Bayir <gunes.bayir@arm.com>
Comments-Addressed: Arm Jenkins <bsgcomp@arm.com>
diff --git a/src/cpu/kernels/CpuAddKernel.h b/src/cpu/kernels/CpuAddKernel.h
index 1afbc1a4..e2062c8 100644
--- a/src/cpu/kernels/CpuAddKernel.h
+++ b/src/cpu/kernels/CpuAddKernel.h
@@ -91,16 +91,16 @@
static const std::vector<AddKernel> &get_available_kernels();
- bool get_can_interpret_inputs_as_1d_array()
+ size_t get_split_dimension() const
{
- return _can_interpret_inputs_as_1d_array;
+ return _split_dimension;
}
private:
ConvertPolicy _policy{};
AddKernelPtr _run_method{ nullptr };
std::string _name{};
- bool _can_interpret_inputs_as_1d_array{ false };
+ size_t _split_dimension{ Window::DimY };
};
} // namespace kernels
} // namespace cpu