Exeptions for not supported converters

Author: Vladislav-Denisov

src/model_converters/tvm_converter/tvm_auxiliary/caffe_format.py

@@ -23,9 +23,11 @@ def _convert_model_from_framework(self):
 
         with open(self.model_params, 'rb') as f:
             init_net.ParseFromString(f.read())
-
-        model, params = self.tvm.relay.frontend.from_caffe(init_net,
+        if self.high_level_api in ['Relay', 'RelayVM']:
+            model, params = self.tvm.relay.frontend.from_caffe(init_net,
                                                            predict_net,
                                                            shape_dict,
                                                            dtype_dict)
-        return model, params
+            return model, params
+        else:
+            raise ValueError(f'API {self.high_level_api} is not supported')

src/model_converters/tvm_converter/tvm_auxiliary/mxnet_format.py

@@ -40,5 +40,8 @@ def _get_mxnet_network(self):
     def _convert_model_from_framework(self):
         net = self._get_mxnet_network()
         shape_dict = {self.input_name: self.input_shape}
-        model, params = self.tvm.relay.frontend.from_mxnet(net, shape_dict)
-        return model, params
+        if self.high_level_api in ['Relay', 'RelayVM']:
+            model, params = self.tvm.relay.frontend.from_mxnet(net, shape_dict)
+            return model, params
+        else:
+            raise ValueError(f'API {self.high_level_api} is not supported')

src/model_converters/tvm_converter/tvm_auxiliary/tflite_format.py

@@ -20,7 +20,10 @@ def _convert_model_from_framework(self):
         model_tf = self._get_tf_model(self.model_path)
         shape_dict = {self.input_name: self.input_shape}
         dtype = {self.input_name: 'float32'}
-        model, params = self.tvm.relay.frontend.from_tflite(model_tf,
+        if self.high_level_api in ['Relay', 'RelayVM']:
+            model, params = self.tvm.relay.frontend.from_tflite(model_tf,
                                                             shape_dict=shape_dict,
                                                             dtype_dict=dtype)
-        return model, params
+            return model, params
+        else:
+            raise ValueError(f'API {self.high_level_api} is not supported')