Add missing axises comment (#3979)

This patch will add missing axises comment.

Signed-off-by: shubham <shub98.gupta@samsung.com>

diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleBatchToSpaceNd.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleBatchToSpaceNd.h
index aced917..5695d97 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/layers/SimpleBatchToSpaceNd.h
+++ b/runtimes/pure_arm_compute/src/internal/layers/SimpleBatchToSpaceNd.h
@@ -33,6 +33,7 @@ public:
    * @param[in]  input       First tensor input.
    * @param[out] output      Output tensor.
    * @param[in]  block_size  Block size.
+   * @param[in]  axises      Axises of rank 4
    */
   void configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *output,
                  const int32_t *block_size,

diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleDepthToSpace.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleDepthToSpace.h
index 0f30cbf..1032aaa 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/layers/SimpleDepthToSpace.h
+++ b/runtimes/pure_arm_compute/src/internal/layers/SimpleDepthToSpace.h
@@ -34,6 +34,7 @@ public:
    * @param[in]  input       First tensor input.
    * @param[out] output      Output tensor.
    * @param[in]  block_size  Block size.
+   * @param[in]  axises      Axises of rank 4
    */
   void configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *output, int32_t block_size,
                  const ::arm_compute::Coordinates &axises = getARMComputeAxises(4));

diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToBatchND.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToBatchND.h
index 500bcd7..4908a2a 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToBatchND.h
+++ b/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToBatchND.h
@@ -34,6 +34,7 @@ public:
    * @param[in]  block_size   Block size.
    * @param[in]  padding_size Padding size.
    * @param[out] output       Output tensor.
+   * @param[in]  axises       Axises of rank 4
    */
   void configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *block_size,
                  ::arm_compute::ITensor *padding_size, ::arm_compute::ITensor *output,

diff --git a/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h b/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h
index fb56034..9e87c36 100644 (file)
--- a/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h
+++ b/runtimes/pure_arm_compute/src/internal/layers/SimpleSpaceToDepth.h
@@ -42,6 +42,7 @@ public:
    * @param[in] input       First tensor input.
    * @param[in] output      Output tensor.
    * @param[in] block_size  Block size.
+   * @param[in] axises      Axises of rank 4
    * @return N/A
    */
   void configure(::arm_compute::ITensor *input, ::arm_compute::ITensor *output, int32_t block_size,