diff options
Diffstat (limited to 'src/core/NEON/kernels/NELogicalKernel.cpp')
| -rw-r--r-- | src/core/NEON/kernels/NELogicalKernel.cpp | 5 |
1 files changed, 2 insertions, 3 deletions
diff --git a/src/core/NEON/kernels/NELogicalKernel.cpp b/src/core/NEON/kernels/NELogicalKernel.cpp index d98694ffe1..e1c24da777 100644 --- a/src/core/NEON/kernels/NELogicalKernel.cpp +++ b/src/core/NEON/kernels/NELogicalKernel.cpp @@ -287,9 +287,8 @@ void NELogicalKernel::configure(const ITensorInfo *input1, const ITensorInfo *in if(op != LogicalOperation::Not) { ARM_COMPUTE_ERROR_ON_NULLPTR(input2); - const std::pair<TensorShape, ValidRegion> broadcast_pair = ITensorInfo::broadcast_shape_and_valid_region(*input1, *input2); - out_shape = broadcast_pair.first; - win = calculate_max_window(broadcast_pair.second, Steps()); + out_shape = TensorShape::broadcast_shape(input1->tensor_shape(), input2->tensor_shape()); + win = calculate_max_window(out_shape, Steps()); } ICPPKernel::configure(win); |