def __update_output_data_from_relay()

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

• 12 lines of code
• 8 McCabe index (conditional complexity)

    def __update_output_data_from_relay(self) -> None:
        from tvm import relay
        relay_output_metadata = self.__get_outputs_from_relay()
        if 'Outputs' not in self.metadata or len(relay_output_metadata) != len(self._metadata['Outputs']):
            self._metadata['Outputs'] = relay_output_metadata
        else:
            for index in range(len(relay_output_metadata)):
                self._metadata['Outputs'][index]['dtype'] = relay_output_metadata[index]['dtype']
                self._metadata['Outputs'][index]['shape'] = relay_output_metadata[index]['shape']

        # Add generic outputs names if none (needed by DLR for RelayVM).
        for i, out in enumerate(self._metadata['Outputs']):
            if 'name' not in out:
                out['name'] = "output_{}".format(i)