diff --git a/openvino_tensorflow/api.cc b/openvino_tensorflow/api.cc index e5e23576d..04f254033 100644 --- a/openvino_tensorflow/api.cc +++ b/openvino_tensorflow/api.cc @@ -30,8 +30,8 @@ void disable() { Disable(); } bool is_enabled() { return IsEnabled(); } bool CheckBackend(const char* backend) { - const char* devices[4] = {"CPU", "GPU", "MYRIAD", "VAD-M"}; - for (int i = 0; i < 4; i++) { + const char* devices[5] = {"CPU", "GPU", "GPU_FP16", "MYRIAD", "VAD-M"}; + for (int i = 0; i < 5; i++) { if (strcmp(backend, devices[i]) == 0) return true; } return false;