Line data    Source code

       1              : // SPDX-License-Identifier: Apache-2.0
       2              : /**
       3              :  * Copyright (C) 2021 Jihoon Lee <jhoon.it.lee@samsung.com>
       4              :  *
       5              :  * @file activation_realizer.h
       6              :  * @date 23 November 2021
       7              :  * @brief NNTrainer graph realizer which realizes activation!=none to actual
       8              :  * activation node
       9              :  * @see https://github.com/nnstreamer/nntrainer
      10              :  * @author Jihoon Lee <jhoon.it.lee@samsung.com>
      11              :  * @bug No known bugs except for NYI items
      12              :  */
      13              : #ifndef __ACTIVATION_REALIZER_H__
      14              : #define __ACTIVATION_REALIZER_H__
      15              : 
      16              : #include <memory>
      17              : #include <vector>
      18              : 
      19              : #include <realizer.h>
      20              : 
      21              : namespace nntrainer {
      22              : 
      23              : /**
      24              :  * @brief Graph realizer which realizes activation
      25              :  *
      26              :  */
      27          697 : class ActivationRealizer final : public GraphRealizer {
      28              : public:
      29              :   /**
      30              :    * @brief Destroy the Graph Realizer object
      31              :    *
      32              :    */
      33              :   ~ActivationRealizer();
      34              : 
      35              :   /**
      36              :    * @brief graph realizer creates a new graph based on the reference
      37              :    *
      38              :    */
      39              :   GraphRepresentation realize(const GraphRepresentation &reference) override;
      40              : };
      41              : 
      42              : } // namespace nntrainer
      43              : 
      44              : #endif // __ACTIVATION_REALIZER_H__