From 471d06512f373fc29f0721f4438117fc062dc6b7 Mon Sep 17 00:00:00 2001 From: Jihoon Lee Date: Tue, 27 Jul 2021 11:41:07 +0900 Subject: [PATCH] [CAPI] Expose layer enums This patch contains layer enums to be exposed **Self evaluation:** 1. Build test: [X]Passed [ ]Failed [ ]Skipped 2. Run test: [X]Passed [ ]Failed [ ]Skipped Signed-off-by: Jihoon Lee --- api/nntrainer-api-common.h | 23 ++++++++++++++++++++++- 1 file changed, 22 insertions(+), 1 deletion(-) diff --git a/api/nntrainer-api-common.h b/api/nntrainer-api-common.h index 19d4ee4..7ee9194 100644 --- a/api/nntrainer-api-common.h +++ b/api/nntrainer-api-common.h @@ -37,7 +37,28 @@ typedef enum { ML_TRAIN_LAYER_TYPE_ADDITION, /**< Addition Layer (Since 6.5) */ ML_TRAIN_LAYER_TYPE_CONCAT, /**< Concat Layer (Since 6.5) */ ML_TRAIN_LAYER_TYPE_MULTIOUT, /**< MultiOut Layer (Since 6.5) */ - ML_TRAIN_LAYER_TYPE_UNKNOWN = 999 /**< Unknown Layer */ + ML_TRAIN_LAYER_TYPE_PREPROCESS_FLIP, /**< Preprocess flip Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_PREPROCESS_TRANSLATE, /**< Preprocess translate Layer + (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_PREPROCESS_L2NORM, /**< Preprocess L2Normalization Layer + (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_EMBEDDING, /** Embedding Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_RNN, /** RNN Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_LSTM, /** LSTM Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_SPLIT, /** Splite Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_GRU, /** GRU Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_PERMUTE, /** Permute layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_DROPOUT, /** DropOut Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_BACKBONE_NNSTREAMER, /**< Backbone using NNStreamer + (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_CENTROID_KNN, /**< Centroid KNN Layer (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_LOSS_MSE = + 500, /**< Mean Squared Error Loss Layer type (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_LOSS_CROSS_ENTROPY_SIGMOID, /**< Cross Entropy with + Sigmoid Loss Layer type (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_LOSS_CROSS_ENTROPY_SOFTMAX, /**< Cross Entropy with + Softmax Loss Layer type (Since 6.5) */ + ML_TRAIN_LAYER_TYPE_UNKNOWN = 999 /**< Unknown Layer */ } ml_train_layer_type_e; /** -- 2.7.4