fix formatting and add comments

openvinotoolkit · Jan 23, 2025 · e3d1155 · e3d1155
1 parent 20218d2
commit e3d1155
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 6 deletions.
diff --git a/src/plugins/intel_cpu/src/nodes/executors/acl/acl_pooling.cpp b/src/plugins/intel_cpu/src/nodes/executors/acl/acl_pooling.cpp
@@ -93,6 +93,7 @@ bool AclPoolingExecutor::isSupported(const TensorInfo& srcTensorInfo,
             if (dataLayout == arm_compute::DataLayout::NHWC) {
                 changeLayoutToNH_C({&indShape});
             }
+            //U32 is specified since this is the only data type supported by ACL
             TensorInfo indTensorInfo = TensorInfo(indShape, 1, arm_compute::DataType::U32, dataLayout);
             arm_compute::Status s =
                 arm_compute::NEPoolingLayer::validate(&srcTensorInfo, &dstTensorInfo, *pool_info, &indTensorInfo);
@@ -184,10 +185,9 @@ bool AclPoolingExecutor::init(const PoolingAttrs& poolingAttrs,
             if (dstTensorInfo.data_layout() == arm_compute::DataLayout::NHWC) {
                 changeLayoutToNH_C({&indShape});
             }
-            TensorInfo indTensorInfo = TensorInfo(indShape,
-                                                  1,
-                                                  arm_compute::DataType::U32,
-                                                  getAclDataLayoutByMemoryDesc(dstDescs[1]));
+            //U32 is specified since this is the only data type supported by ACL
+            TensorInfo indTensorInfo =
+                TensorInfo(indShape, 1, arm_compute::DataType::U32, getAclDataLayoutByMemoryDesc(dstDescs[1]));
             indTensor.allocator()->init(indTensorInfo);
             exec_func = [this, pool_info]() -> std::unique_ptr<IFunction> {
                 auto acl_op = std::make_unique<arm_compute::NEPoolingLayer>();

diff --git a/src/plugins/intel_cpu/src/nodes/executors/acl/acl_pooling.hpp b/src/plugins/intel_cpu/src/nodes/executors/acl/acl_pooling.hpp
@@ -79,8 +79,7 @@ class AclPoolingExecutorBuilder : public PoolingExecutorBuilder {
             return false;
         }
 
-        if (dstDescs.size() == 2u && 
-            (dstDescs[1]->getPrecision() != ov::element::u32 && dstDescs[1]->getPrecision() != ov::element::i32)) {
+        if (dstDescs.size() == 2u && !one_of(dstDescs[1]->getPrecision(), ov::element::u32, ov::element::i32)) {
             DEBUG_LOG("AclPoolingExecutor supports U32 as indices precisions only. ",
                       "Passed indices precision: ",
                       dstDescs[1]->getPrecision());