diff options
Diffstat (limited to 'runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h')
-rw-r--r-- | runtimes/pure_arm_compute/src/internal/layers/FeatureLoggingLayer.h | 25 |
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()) |