def METHOD_NAME(self, feed_dict=None):
    if feed_dict is None:
        return super().METHOD_NAME()
    return {
        input_name: tuple(input_data.shape)
        for input_name, input_data in feed_dict.items()
    }