[Filter/TFLite] code clean

move unused method to private and remove extern methods

Signed-off-by: HyoungjooAhn <hello.ahn@samsung.com>

diff --git a/gst/tensor_filter/tensor_filter_tensorflow_lite_core.cc b/gst/tensor_filter/tensor_filter_tensorflow_lite_core.cc
index 34dd071..6587fdc 100644 (file)
--- a/gst/tensor_filter/tensor_filter_tensorflow_lite_core.cc
+++ b/gst/tensor_filter/tensor_filter_tensorflow_lite_core.cc
@@ -438,30 +438,6 @@ tflite_core_getOutputDim (void *tflite, GstTensorsInfo * info)
 }
 
 /**
- * @brief      get the size of Input Tensor of model
- * @param      tflite  : the class object
- * @return     the number of Input Tensors.
- */
-int
-tflite_core_getInputSize (void *tflite)
-{
-  TFLiteCore *c = (TFLiteCore *) tflite;
-  return c->getInputTensorSize ();
-}
-
-/**
- * @brief      get the size of Output Tensor of model
- * @param      tflite  : the class object
- * @return     the number of Output Tensors.
- */
-int
-tflite_core_getOutputSize (void *tflite)
-{
-  TFLiteCore *c = (TFLiteCore *) tflite;
-  return c->getOutputTensorSize ();
-}
-
-/**
  * @brief      invoke the model
  * @param      tflite  : the class object
  * @param[in] input : The array of input tensors

diff --git a/gst/tensor_filter/tensor_filter_tensorflow_lite_core.h b/gst/tensor_filter/tensor_filter_tensorflow_lite_core.h
index 3f7c465..98ed2a3 100644 (file)
--- a/gst/tensor_filter/tensor_filter_tensorflow_lite_core.h
+++ b/gst/tensor_filter/tensor_filter_tensorflow_lite_core.h
@@ -49,8 +49,6 @@ public:
   const char* getModelPath();
   int setInputTensorProp ();
   int setOutputTensorProp ();
-  int getInputTensorSize ();
-  int getOutputTensorSize ();
   int getInputTensorDim (GstTensorsInfo * info);
   int getOutputTensorDim (GstTensorsInfo * info);
   int invoke (const GstTensorMemory * input, GstTensorMemory * output);
@@ -65,6 +63,8 @@ private:
   std::unique_ptr < tflite::Interpreter > interpreter;
   std::unique_ptr < tflite::FlatBufferModel > model;
 
+  int getInputTensorSize ();
+  int getOutputTensorSize ();
   tensor_type getTensorType (TfLiteType tfType);
   int getTensorDim (int tensor_idx, tensor_dim dim);
 };
@@ -82,8 +82,6 @@ extern "C"
   extern const char *tflite_core_getModelPath (void *tflite);
   extern int tflite_core_getInputDim (void *tflite, GstTensorsInfo * info);
   extern int tflite_core_getOutputDim (void *tflite, GstTensorsInfo * info);
-  extern int tflite_core_getOutputSize (void *tflite);
-  extern int tflite_core_getInputSize (void *tflite);
   extern int tflite_core_invoke (void *tflite, const GstTensorMemory * input,
       GstTensorMemory * output);