def _operation_to_pooling_shape(operation):
    """Takes in the operation string and returns the pooling kernel shape."""
    splitted_operation = operation.split('_')
    shape = splitted_operation[-1]
    assert 'x' in shape
    filter_height, filter_width = shape.split('x')
    assert filter_height == filter_width
    return int(filter_height)
