[ML][Training] Model implementation

[platform/core/api/webapi-plugins.git] / src / ml / ml_utils.h

diff --git a/src/ml/ml_utils.h b/src/ml/ml_utils.h
index 5deae9c..ccfeb16 100644 (file)
--- a/src/ml/ml_utils.h
+++ b/src/ml/ml_utils.h
@@ -45,10 +45,9 @@ extern const PlatformEnum<ml_nnfw_hw_e> HWTypeEnum;
 extern const PlatformEnum<ml_nnfw_type_e> NNFWTypeEnum;
 extern const PlatformEnum<ml_tensor_type_e> TensorTypeEnum;
 
-// MK-TODO implement internal enum or remove from API design if not needed
-// extern const PlatformEnum<TODO> DatasetTypeEnum;
 extern const PlatformEnum<ml_train_optimizer_type_e> OptimizerTypeEnum;
 extern const PlatformEnum<ml_train_layer_type_e> LayerTypeEnum;
+extern const PlatformEnum<ml_train_summary_type_e> SummaryTypeEnum;
 
 }  // namespace types