Codestyle

Author: Vladislav-Denisov

src/benchmark/frameworks/tvm/tvm_parameters_parser.py

@@ -53,7 +53,7 @@ def parse_parameters(self, curr_test):
             optimization_level=_optimization_level.data if _optimization_level else None,
             layout=_layout.data if _layout else None,
             target=_target.data if _target else None,
-            high_level_api = _high_level_api.data if _high_level_api else None,
+            high_level_api=_high_level_api.data if _high_level_api else None,
         )
 
 

src/model_converters/tvm_converter/tvm_auxiliary/caffe_format.py

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

src/model_converters/tvm_converter/tvm_auxiliary/converter.py

@@ -184,7 +184,7 @@ def __get_graph_module_from_relay(self, mod, params, target, dev):
             lib = self.tvm.relay.build(mod, target=target, params=params)
         self.graph = self.graph_executor.GraphModule(lib['default'](dev))
         return self.graph
-     
+
     def __get_graph_module_from_relay_vm(self, mod, params, target, dev):
         vm = self.tvm.runtime.vm
         rly_vm = self.tvm.relay.vm

src/model_converters/tvm_converter/tvm_auxiliary/tflite_format.py

@@ -22,8 +22,8 @@ def _convert_model_from_framework(self):
         dtype = {self.input_name: 'float32'}
         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)
+                                                                shape_dict=shape_dict,
+                                                                dtype_dict=dtype)
             return model, params
         else:
             raise ValueError(f'API {self.high_level_api} is not supported')