summaryrefslogtreecommitdiff
path: root/runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h
diff options
context:
space:
mode:
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h')
-rw-r--r--runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h25
1 files changed, 25 insertions, 0 deletions
diff --git a/runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h b/runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h
index 502a1ee0e..83ae7c17b 100644
--- a/runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h
+++ b/runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h
@@ -14,6 +14,12 @@
* limitations under the License.
*/
+/**
+ * @file        FeatureLoggingLayer.h
+ * @brief       This file contains FeatureLoggingLayer class
+ * @ingroup     COM_AI_RUNTIME
+ */
+
#ifndef __FEATURE_LOGGING_LAYER_H__
#define __FEATURE_LOGGING_LAYER_H__
@@ -27,9 +33,24 @@
#include "internal/arm_compute.h"
+/**
+ * @brief Class to run FeatureLogging Layer
+ */
class FeatureLoggingLayer : public ::arm_compute::IFunction
{
public:
+ FeatureLoggingLayer(void) : _tag(""), _target(nullptr)
+ {
+ // DO NOTHING
+ }
+
+public:
+ /**
+ * @brief Configure the layer
+ * @param[in] tag Text tag for this layer
+ * @param[in] target The feature tensor to be printed
+ * @return N/A
+ */
void configure(const std::string &tag, ::arm_compute::ITensor *target)
{
_tag = tag;
@@ -37,6 +58,10 @@ public:
}
public:
+ /**
+ * @brief Run the operation. Must be called after configure().
+ * @return N/A
+ */
void run(void) override
{
if (::internal::arm_compute::isGpuMode())