def load_model()

in src/neo_loader/tensorflow_model_loader.py [0:0]

• 17 lines of code
• 3 McCabe index (conditional complexity)

    def load_model(self) -> None:
        self.__extract_model_path_from_model_artifacts()
        self.__tf_model_helper = TFModelHelper(self.__model_path.as_posix(), self.data_shape)
        self.__extract_metadata_and_output_tensor_names_from_model()
        self.__extract_tf_graph()
        try:
            self._relay_module_object, self._params = relay.frontend.from_tensorflow(
                self.__tf_graph, shape=self.data_shape, outputs=self.__output_tensor_names
            )
            self._relay_module_object = self.convert_layout(self._relay_module_object)
            self.update_missing_metadata()
        except OpError:
            raise
        except Exception as e:
            logger.exception("Failed to convert tensorflow model. %s" % repr(e))
            msg = "InputConfiguration: TVM cannot convert Tensorflow model. Please make sure the framework you selected is correct. {}".format(e)
            raise RuntimeError(msg)