From 160ace3c15d08a5f3146393b4fdc57ae1fdc39f6 Mon Sep 17 00:00:00 2001 From: Jaeyun Date: Thu, 2 Jul 2020 11:40:22 +0900 Subject: [PATCH] [Filter/TF] code clean change to static function to release buffer. Signed-off-by: Jaeyun --- ext/nnstreamer/tensor_filter/tensor_filter_tensorflow.cc | 7 ++++--- 1 file changed, 4 insertions(+), 3 deletions(-) diff --git a/ext/nnstreamer/tensor_filter/tensor_filter_tensorflow.cc b/ext/nnstreamer/tensor_filter/tensor_filter_tensorflow.cc index 3fa7695..f9c6025 100644 --- a/ext/nnstreamer/tensor_filter/tensor_filter_tensorflow.cc +++ b/ext/nnstreamer/tensor_filter/tensor_filter_tensorflow.cc @@ -95,6 +95,7 @@ private: tensor_type getTensorTypeFromTF (TF_DataType tfType); TF_DataType getTensorTypeToTF (tensor_type tType); int validateTensor (const GstTensorsInfo * tensorInfo, int is_input); + static void releaseBuffer (void* data, size_t t); }; void init_filter_tf (void) __attribute__ ((constructor)); @@ -192,8 +193,8 @@ TFCore::getModelPath () /** * @brief the definition of a deallocator method */ -static void -DeallocateBuffer (void* data, size_t t) { +void +TFCore::releaseBuffer (void* data, size_t t) { std::free (data); } @@ -232,7 +233,7 @@ TFCore::loadModel () TF_Buffer* buffer = TF_NewBuffer (); buffer->data = content; buffer->length = file_size; - buffer->data_deallocator = DeallocateBuffer; + buffer->data_deallocator = releaseBuffer; graph = TF_NewGraph (); g_assert (graph != nullptr); -- 2.7.4