[nnstreamer/backbone] Update to support more backbone
authorParichay Kapoor <pk.kapoor@samsung.com>
Wed, 4 Nov 2020 04:00:11 +0000 (13:00 +0900)
committerJijoong Moon <jijoong.moon@samsung.com>
Thu, 12 Nov 2020 11:46:17 +0000 (20:46 +0900)
Update nnstreamer backbone layer to support more backbones than just tflite
Wait for this PR to merge till the bug fix on nnstreamer
https://github.com/nnstreamer/nnstreamer/pull/2850 is merged and reflected

**Self evaluation:**
1. Build test: [x]Passed [ ]Failed [ ]Skipped
2. Run test: [x]Passed [ ]Failed [ ]Skipped

Signed-off-by: Parichay Kapoor <pk.kapoor@samsung.com>
nntrainer/layers/nnstreamer_layer.cpp

index 38150318dbcf7c39639847625ab43e0f653ad657..9b78287329cb897f0a55ba4f5ce3ae6642461e36 100644 (file)
@@ -96,7 +96,7 @@ int NNStreamerLayer::initialize() {
   TensorDim in_dim;
 
   status = ml_single_open(&single, modelfile.c_str(), NULL, NULL,
-                          ML_NNFW_TYPE_TENSORFLOW_LITE, ML_NNFW_HW_ANY);
+                          ML_NNFW_TYPE_ANY, ML_NNFW_HW_AUTO);
   if (status != ML_ERROR_NONE)
     return finalizeError(status);