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 flatten_realizer.h
       6              :  * @date 09 October 2021
       7              :  * @brief NNTrainer graph realizer which realizes flatten=true to actual node
       8              :  * @see https://github.com/nnstreamer/nntrainer
       9              :  * @author Jihoon Lee <jhoon.it.lee@samsung.com>
      10              :  * @bug No known bugs except for NYI items
      11              :  */
      12              : #ifndef __FLATTEN_REALIZER_H__
      13              : #define __FLATTEN_REALIZER_H__
      14              : 
      15              : #include <memory>
      16              : #include <vector>
      17              : 
      18              : #include <realizer.h>
      19              : 
      20              : namespace nntrainer {
      21              : 
      22              : /**
      23              :  * @brief Graph realizer class
      24              :  *
      25              :  */
      26          697 : class FlattenRealizer final : public GraphRealizer {
      27              : public:
      28              :   /**
      29              :    * @brief Destroy the Graph Realizer object
      30              :    *
      31              :    */
      32              :   ~FlattenRealizer();
      33              : 
      34              :   /**
      35              :    * @brief graph realizer creates a new graph based on the reference
      36              :    *
      37              :    */
      38              :   GraphRepresentation realize(const GraphRepresentation &reference) override;
      39              : };
      40              : 
      41              : } // namespace nntrainer
      42              : 
      43              : #endif // __FLATTEN_REALIZER_H__