From d7d7e9035ca28b1b5200b20a73825397d46830fc Mon Sep 17 00:00:00 2001 From: Georgios Pinitas Date: Wed, 18 Dec 2019 15:40:54 +0000 Subject: COMPMID-2819: Update data type auto_init in Elementwise kernels. Auto initialization functionality is updated in elementwise kernels to check both data types in order to reason for the output data type configuration. Signed-off-by: Georgios Pinitas Change-Id: Ic08b5567d08a3aaca00942acbdbc8aee19686617 Reviewed-on: https://review.mlplatform.org/c/2495 Tested-by: Arm Jenkins Reviewed-by: Giorgio Arena Comments-Addressed: Arm Jenkins --- src/core/CL/kernels/CLElementwiseOperationKernel.cpp | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) (limited to 'src/core/CL/kernels/CLElementwiseOperationKernel.cpp') diff --git a/src/core/CL/kernels/CLElementwiseOperationKernel.cpp b/src/core/CL/kernels/CLElementwiseOperationKernel.cpp index a69e84a16d..66f7ee0a56 100644 --- a/src/core/CL/kernels/CLElementwiseOperationKernel.cpp +++ b/src/core/CL/kernels/CLElementwiseOperationKernel.cpp @@ -195,7 +195,7 @@ std::pair validate_and_configure_window_for_arithmetic_operators { set_format_if_unknown(output, Format::S16); } - else if(input1.data_type() == DataType::F16 && input2.data_type() == DataType::F16) + else if(input1.data_type() == DataType::F16 || input2.data_type() == DataType::F16) { set_format_if_unknown(output, Format::F16); } -- cgit v1.2.1