diff options
Diffstat (limited to 'src/cpu/kernels')
-rw-r--r-- | src/cpu/kernels/CpuAddKernel.cpp | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/src/cpu/kernels/CpuAddKernel.cpp b/src/cpu/kernels/CpuAddKernel.cpp index 85ae410a94..8fc82ec184 100644 --- a/src/cpu/kernels/CpuAddKernel.cpp +++ b/src/cpu/kernels/CpuAddKernel.cpp @@ -41,7 +41,7 @@ namespace kernels { bool can_interpret_inputs_as_1d_array(const ITensorInfo &src0, const ITensorInfo &src1) { - return !src0.has_padding() && !src1.has_padding() && src0.tensor_shape() == src1.tensor_shape(); + return !src0.has_padding() && !src1.has_padding() && src0.tensor_shape() == src1.tensor_shape() && src0.strides_in_bytes() == src1.strides_in_bytes(); } namespace |