def METHOD_NAME(self, new_shape):
    '''
    Reshapes the model inputs to fit the new input shape.
    Args:
        - new_shape (dict): the dictionary with inputs names as keys and
            list of new shape as values in the following format:
            {
                'input_layer_name_1': [1, 128, 128, 3],
                'input_layer_name_2': [1, 128, 128, 3],
                ...
            }
    '''