From: HyoungjooAhn Date: Mon, 26 Nov 2018 10:45:00 +0000 (+0900) Subject: [Filter/TFLite] code clean X-Git-Tag: v0.0.3~36 X-Git-Url: http://review.tizen.org/git/?a=commitdiff_plain;h=e0d610520017207b450fc7a5ceb9ffad7c843a00;p=platform%2Fupstream%2Fnnstreamer.git [Filter/TFLite] code clean move unused method to private and remove extern methods Signed-off-by: HyoungjooAhn --- 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 --- 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 --- 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);