}
/**
- * @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
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);
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);
};
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);